/*
* 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 = 2516;


static const batchedGemm::BatchedGemmConfig tllmGenBatchedGemmList[] = {
#ifndef EXCLUDE_SM_100
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "c3d504866d6ce842acac317e0af1393505e209b3123eaed5259c42d60ca8d9bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "7840fb4c57f0e90f5f38da46fb4afad96d675e9ca2f937cbceac651999cc2c71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "ebeb1e6b5fe9e05b8a537b1a8cb4ae3b2953b3a09736a2387f7332a6e443247c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5391df66e6a423e4cdbc6f07499d3de7032478ce696158f8a9a9d1e6403066f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_bN_rgTma_clmp_sm100f", 256, "d58d898f4e44b0a2bdc870940b44b0a76986d781839cfecae2cdfd4768180d02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_bN_rgTma_clmp_sm100f", 256, "f68022cede4275e055f89534a634a19c1b29651d6256bc1c42b49ace9e832d8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "0d53cbaf71c262d57e6ed77454b2c7a941cf3154ed95680396d01bd1875275a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e92e4013aad097c8d4b9fd0b0e7606983bc6863784e8da6cc8f08f9f74abcb47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "8b77a91c4498b5d9b3577192da7595071905d5846d9f6c15c4dd58d7e9d37b19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_bN_rgTma_clmp_sm100f", 256, "7a047d812ae9397fc6325c62f3c3b163c9acdb60c7cfd18548d2f109d7467ffd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "d565b039c09060ad4736034e36035e403b73680712030582e14d372379c404c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e87f3bd3e5a3c4d60472337e8e7af1662697807bf86899dc363700e5193f9e0a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "6589b01553cddc02fa1dac511e7055a2cc615898118d462234a078a75de1581f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "d65a458a98ce8dde2c7d84ca0b68bc606c7844da7eb4cd12b4fb46fdef5b78d5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "4ea4d3048da86dafb2482debeab441c6324555c6630a9c1ae04dfbc7b408c736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "60906813ee08ceafadd76cfd4a75155b39d61e102468eae5c8df116b92a35ade", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "ced02fe9d6f5d3c79a354f4fbe4b8aa79c5b3364a31294422125fe3869b054b0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "cbeb0867af7a1399974e711cd12cf7737284ca6654593181a50b94ac6539d207", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "47e74d1ee15f067d7a3eb272d151057563794e048235b1507dffa776ea9366bd", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "4d62dcfbc6ce449dc24970401dda374e29ef30128e4b8d1b44ad6a360924d77f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "8f8eac52152e6581a405f206726ec7b5bf1f53eb0e2dddc61e87a002ca7252e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "a379284566537bcff8c8df679a73c6067350833062c5125ba33d9c0eb100c386", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "42a5dd0aac72877ca90ccd49330758dd0efe092256668e1361fcd50fcd779194", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d6b9bd8ae67f74b7adc80cd8512f2119bf446831fae056f98a50ae8e9ff39bed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "954dcf291da3db91f04a2dff978a33602681dabb9bb61c8aeba793d1967c8a02", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "4d554dad29b28cdf3baf5142f6fa46e03000799f66d84795719b739f4ae4badb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "9451f4076cc981c2cc24505fadeb5a102c5520eceb50d75fd083f2d49aadb3a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "3417df658bb40c3672e6ac49983fbd641d0fc71e6e205281792bbd18428979fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "38a4baaa90c95e7a4260a5e6f06df641cc94caaae4b37522a8b7131c94d118ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "4e59f5ad71acf29ebde083aef8bf4721ad7034a1ced1eb4f3bfdff22b83412e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "3aa68b21206a252e0508c01cfe917f088b6a84637dcc62ef118cf6e533fcbf87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "cc530838bd8c11d3e36a1e204c0b117b92d361dd5ab09bcaa2fca7d25806e209", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "2451d005881a51eed80528cb2bbe53ef8dbeb03fd232127de12a0cf9043c2ae0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "30172aeef1c37849279b3cf9618f6ffdc7fa1cb02d31929b2cab068f84ef7d50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "08217cab34f0381785ec508078a6630a4e57c31bd6c86bc13235c639578b7cf6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "1e52626e3c41dd5fc4d0b1c655c938863a05ca69c26acaa393b60ac90af67918", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e3971ee5bd4f864b8168e15e6e63b42a0f6bbc9fc7d82c8efbafa7900c0e9432", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "999cdd30a464b1df618a41249183f2881b88d7147f6f28b9b40a4760e239b103", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "452daa886b383022f8a79bcc81cacd228a73b7b2c6ef54b6a13223608a804e40", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "208da1e0b6a81042b19ee58c9ef4394c3a44f85a777bbd151efb15305409148f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "9d886c5967712f0a990af41159783e992eb1c2202b1f892947ace5087fb4e50a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "659db3d7c11f42a28479bc01584cadeb15afe3a9380e3b604f596ec17925a752", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d02772b77cbf555e9294b9f56147a63fee1d2d1eb74958cb701f4b16538bfc97", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "c0c36c695f7bcbf7bd25944df56d262aebe65b5f98e771bab72e145c98fd599a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "7b29562a359f046965b09ff09f9ef11b0e75e757a3eef4b6ed82e078be9682bf", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "94c71262d7e5ab4c32826b0c2300331d7a2af818670fedd0dcf98546a85d2577", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "306a9b2fc42a8dba900b841065b84843344b046133e851640b248ccbd349193e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "7554bafa4e0e7f4c1274f3f6db4e70c5189869bff25f01301758e716616b0047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "482f267b871452f6a776a17c64a17c774a92a3861ba2ca0034a0e564ece633c3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "57d948d73e7c54b153777490a94f902d37a3d15859e99c5895d0d0cc44fe5faf", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "afa97729c83a60b5d334caa2b61415c7096280ed5e0ec4ed3b584d61dafd6901", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "6c2acacd475cd3d8d37b71b60e9388419fe73c2030c1a9f1c59bd14707430260", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "2774ad813f174beb3262ac92e9bb0a8e44305178f57c6e7906f49406950a0a37", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "0d1dab100b984ed6aeae5923d0bc9d155fc5f296d3bd1e952a0f1e3742e7ca00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7924d925ed6696c27d138eb7c5c7bc0df7d59784332a5230dbcc6fe82f6b13a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "f3e2b93a94b97e4261927b498aa634073187fb84b2d3baa76dc3a14a89f7e3f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "e8ff304b628b91fadc7decdd3b00e21cb81f917ba514d833b3af944a3f51231d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d445680d7056a3b1d007e7b1ac0d91afde4d0c6f11b11b2ad1d064b8e76c783f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "04d7205c4fc40779eaa6e69594bc4af594247acae0074aa0485a9c2c82a5afb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "18a4ec3553a1182375c762bdedb5eda8c2e56723c4ef0928bf1e75a11c2cec10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "1950a457741d1a38a5d23d4015d0675d12d0fabe6dbc70f797ec496511d8c09b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "edf4f4ddbf17ec01efe05105e2d033c56372e487c4c30a6eb987a2ae40db35d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "23d89e0e7c07be8d4108d9a24ec5fd158f4990e9e83b1d03689fc5d71aa8ca75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c746539b248df8f09e670e56d7d42b3715d287bfc4678e5dabfb867941416f3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5af192df16b64d3ae3d3034b8b7bc4d223e3f6c48d1c3280f892685fe8fcf893", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "bd5cd058667406dc8eb1a3d5bf49ac1c4f9a8eefaf3cfc50eca8ebcfc159a080", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5939afabe08db2d10a78c04da7f9a9ad382983a6054703562e259eb7b3e90434", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7e5426b208365fe0080a02d1f1337f394226f8aaffbf4e05017d30816f8e53ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7ef417ab88c75eed93e5546def461b2f1d55ae01f8a28cc243b8b4499624f6a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "4da91680ef5e77021d1cdf2d1d76f6ca971a3626efa6ec31529377bfb934299e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64c3538356409805b5f2841e7958dabab01740f16214ca74d620a71da0fa081b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "25ed85394bc041e1ae8c8a7af9cfca84021a134991237c9b9c59c9a6d65a0ff4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "818b79b65ac9e05f030beb68592ecdb736389c3af59d82c7996985b734287af7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8d8cc21356f2c012fd5bfc913cea82695a60fdbbd1b6b6f2e0b9d9fc1ab8d830", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "73bd50508301aed95430b44b9e9bd6107d3bd4d27df702cbbf9770f810c58237", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3cfbb203dee6fabcefd9a07d0fb5811f581345aa5ad3fd0804a592d729a0d24a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "450333dbf4f287b45d7d4c143fe2643466587e5b739a1cc23315932f19285254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "7515a967176035ee58508318cc2de006cff4d15cfca88a25e8f962721766b426", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "cce237e8cc27555dffaac32f7fc7e0cdb0a74e4be387164c1c1f0d2bdb346d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "447989021dddcfb91fabdd3e6f39ff535c0979d077cf63a49603d51fe2522e65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "7ecf8177f7d59af9da7b3bc7c272b6067b97bca5a969514d5f4fc19937f18ef4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e67bdbfc38454b9d2e3075ad3a85d4d7488cca343f4039c2f1bcc95bae6adf6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5eeb6200dc9783e10a7cc244d122655d60992edb47ed09cc5318c6e944489673", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0aafdf42e6b1b3329fd1e23309254952408aa8ba7cbfb476af129a73181d49e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a1960c217fb106fad0431c18569dde09ac3ab4d0916e0b5efa0c100a7d041b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e8eaa95721afcf5fee4438715a0673b55ada687d59c1b8fbc9aa3bd7b8e6b268", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f4b1483b9e20968c1e00266a6ba8bf55e1880c7e767dd813254337a98e85f87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e4ae5422e7bdedd3f57a218d38b6f690a19441fdce8715976423746092b5c0e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44331e7ef5571c73c271ce80a7d6da298536a6d64dcca13a5420169993abc4f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8243ade30620387477282b3d2c3b7bd10e6905a2654502539e27760d6cb0312d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "31686d58abf2be673911f9ab94d7b16b6bdfa4af75767f03ba624049d5cb1cf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "90d0733e05b388b9f2ba0e5a0bcdb197245720744e1a46bfb9716d5755df2a35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "19b660fbbe1512233b3cb2ac9de49f414849a3f099bfac0781eb14756b529457", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "007c59858b360ef130addbc9828a5edc38d79e3c0ccad6defdf54b3de4df3fe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f9d76cc150534f2eeddfe2cbec2560d9fed32959f9fda0f7ddfdd17944a61a43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "67e8011cec0b809d30dedaafeed6f9193d518cec7b104d891cf1ea769eedd87d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "6cd3186b6733c3439e35c95a52d4f9eacf76aa164e2221b0b221a52e339ed4fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3376c5b68be03d1ad7bb6c846220958f9a4810b1bf296c6496f640529cc94266", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "be2098a434d8708ad8359da000fd921f6ad9c6ba487bfadbe3a55b0e166ee219", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5a5aa6a5050d051cd236314e7b90101a40d2e3113aaba25278565986eb921e4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "a9915c2fceb106cfca2a28ce7d5c68ecf474c4159fddf0cfa317e3e519e56cdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f97a36f6032c031d0f21309502a45674ed62c81353d6b6fba671617c447a108d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5209ab4ec158b758526e14de10c142a0f0dd4f818d344695e9e46ed173f047ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e8c2c532f7131b95c652aab0067e9cd44ed386bf5a67e39f6cc6fef7d657e361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "b02eb57ec644b876953317be225f47ea3ec39522b9b451d7de2f8c8838dc61a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "8cc375ef8349c137f8afc1dc4e6f4a638fbce710ebeb300415be46be78c9b99a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8f93acf003308758bae3b2481d1a7a2725d28d69351b8f16d8adcc63acdf71c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "68059b2ae191522da29b44d2c47a8f0cc6c0ea24eebdadd989a99e513778529b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "36ef21bfe562c86337ba3d91b2b01fdd699123e80f9dec5c545ec75d8d568814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "660b76bc3827dc497d732ad8c4831f5e5b700a96f6d293c58631194a522d74a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d760345547eacdef0e9e0c22489158475b96362cc024c8f8a2331f2420700055", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "997e8e0c15cb1ca64a3e43ae93cf03f95aff127c1f9a73a0d83a254a76b16a59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d0ea5f0c8461c4cc0e5365eb56133c21ddea45afccaad82c43c76cd5efc9b1ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "8eba9e55a732f419ea16862b3e8991ea6f6b737020ab6605ee9f2e9a3c1a7164", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "a787d79dd00ab2d425e5e829a5239ce9670a250dd255323809ec21b14968e010", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2579e8ef893d6ca712a11b908aaf9b9f21cdb2372863bab4440bb8c8478213a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "abec78b4eda2a42593ed5cd45a8e66c4888204bdb5ef6ea67e21892bed435e32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "8c7f5416b3ebb81d53eeec9a389c40ecb8e595bf3e3358567a33a972616e2ee9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "377b1679cdd16fd8ca4c51c50bac3f278bc433999c64c8809228dcec0fff8810", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "8e3b7bc68e096671a1d0c9a104c9b7dc67c5bf1412b50fb9368ff34e01ca2d43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e6de926cde4211e754a404fa5d234bec05501f752f9100763bce4fdbc036c845", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f45fcd60305dd02636a3051b5fa93e5e55f9a087d6412566f4e4042c3394fa84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_swiGlu_dynB_sm100f", 512, "88908e359689cf20b9ccd33eed2de9cae3ddd5ca1001e194d71eda57d181c9a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "60ea0c62f5c325ec437d37f65f06989c72d419e1641f9cb575de9a9d92246603", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "39d9b9ae595c843434514b40ae2f495252ad7522d825c2ef57c952192b30278e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e50dd36acb365808a629a6652ad70dfce627b745e1672e2d99dfe81bf844289a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "79ca05084f4deb6938ddcd94b6c763170092c24bc7aa094ed9b51552b91e0cc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e50faa70e1230b29320319dc0da6b519b9d3e52f555229adf232b111f16a5ed5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0c6740d626448c3c7ac0fa23ea1d493bcb95ebe2383d819afb5ac65c7c411c83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "87ed4dd67ba1876dececa345786ca5f902fafa98501ff6c316d4ed2c52171791", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7163f019c4aa8ad2318606a6fcd77e7e5d5c18777ba6aa612fe4bfe2b90910e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "7e274e7e0afc65892f8e68e8d05ff2d52ce9f4f8d8ab1c314646ad299798f048", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "54a0ec51a7ec36a7786cb36fcfad38448490f9d05f87c9a4e6fc5c48811ac8e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e748326ce25ccbbc27f631ccedce9841f0b110ac9f0a7ffff1be113339ae8cf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d69e5ee841df8982f8b5a7aa7c063e3717b939d2be42d66f44d89aa53739e383", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3add4f73becd91fedeb697331585cc4ebb46acf3d0741f55fb7e33d6034ede66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "2e8a6ffda2d25ad72b367e07a5f686819c355a5d66ad8e836432d53251701dec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "1ced4d461681f4e945fbcc5a9220de011284301b72320bff13fda846b904d28c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e2d28216982ce3d8452da2d4df50993d89e73b4db04ef5bbad11414fb4e2a1b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 512, "7971315d26c1603b92b654ea2750041dffb1b1533c41825efa807637cfae8a27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d424534a6e6d2dfda62f570f9c6f67280749583fdd113505a9c83a0e19fac5d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "44c0d4c91ba6c04af71dc21f29b48547dbf9630aa779e6e831698050441e7d4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "854aff35744d40dce058ffe7032c268e71380083bcb672eba4574b67af7a8e56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "6adcdb02026f4b13b237cd77e96d78f23c77c111787ecc3ff5bbd64a42d71205", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d8fd294fd3f4a0b108597a787cddf510858cf7bd692b84eaff3700d540d8c1c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f2476f5f100fa02cdbe07de003ba90b8b182bef287df4a471048ff40f8919759", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "630cc01de68becd27dccfe3156a1a340f5604a247f36fe3c40cff2adcb04c9df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 640, "8c4e21edf4c9fea26924e6637a90b10d16f1b4c5ca322d752a5530b57ce47557", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c1b794d1185773d9ebd1a84bdc1ec1c5db4207e5cbd2f42b0e2ed0eff3dc8283", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "475ad3d13793d4fe20ba3e808d151ebcfc4de9ee2e6e3fc7bf4fd5876cbe253e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "dbec557812b9ae5e36ddc99fe03220698bdede83e549ac63c98942dc5e00671f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "f64b203cb82355453585ae2814af8c35532e09338b20eeee0e8d8e2c0c9bdb78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8f5f2cfd4d30f68213230421fad4af63e2a72b6bdc1164c758e65cdb93e39ede", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "4ec655e92735b9861cfc10fc835a77128267efb25bdef46f5d918b1050b89855", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "90e472a967a170ecee666f3cb95a6d9c0edeac664378436e181799e01f8581c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f1bc06b5afe2802b3c217606a790c8e375c4fd6e36d972257182d6d71b2b3a8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "73400b073d7f29ac961636f97bbe5bc8575093204135b2eb24f74aca1bb3b0d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 640, "3d6b18897b97719582d876095683b2ae8e551fb5fb838cb2077d40ee4da35805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "21320f38b70e51935328aa9cd1535d4c6ea87639c1f1f62fb1b2d17fd8a442a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "4c63634a2b3879bddb8c917bbaa4c1db9a911f248a37094d8e9eb333ba02ec0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "96d41fe1b4bb2963f1c3df46a15f5a718a6f6a90f83e70181ae0fa1efcee51c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "574ff22b4b2fca97e117795c235e67d3e39017a646600cd5ea889a5355a2f850", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ae392dc094a00e775a7f29fb05167313307230b33638e13ef2f303419380ae92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "3bbc69437a59f3051eb1ab7eb9240dfda6a094b0d6780ecc332e76ed5962159f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_geGlu_dynB_sm100f", 512, "b92138a4344c7168993a0cc80b118e4f2daaa1d1afaaa186ea5c999698a7006a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "84c7d73e887b2d77f83582e27bc615c9349fcde9cdafd0e8f854afcd4cf71a95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "3241cdf1ffa5cc0fb5f5bb4d5b9740ec435312c4a5ea14690909e031517726f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "bf15d0c1b13bb11a12b7afda323644c054e1d1451b3b5d69325a776e4a1271bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "70b2b291284363323438dadb3fff766eb71cd3c0c240deef30633a2fa0456eb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "35d237b7b8f0ddd43f649a79e65d3e31e7cc21ecbc27b3960acf9dc76b059c86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_geGlu_dynB_sm100f", 640, "49fbda52bc07bdeb6da792cbb8dbddf2923086c62323a9236f51d95a9bb5aedb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "876d5d52c75825914900ffcbd6ec3ab278652d03a68d10c2722536477d8be950", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d0c407b98138e989d2fc9945ff9058d61b7dec7e7100bc5c1f850e6c6e350674", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "9bcbef3061e51456b61c563763d33eba320ee674992494e9a6797c88dc74b22e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c331a503558a88a21062a7f2ce2babace277fd575fa44eea52d6f6dc002322fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0bab8b02a38b67c2a7e3f9b057b3472a83fab836d256ca89718dd13c5ee8b4dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fc5fdd65e1283eea50d0c481802f62fbce9cae9fb654ed8c09b196243f5d1e0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "8c27a53aa3b0aed9aed65ab4ea324b63fd7c278442c12054df81990f2f741607", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "1812301f8262f8de4e8928df3a3c27161b04fe15135ffa9f9bdc986cdea59ffa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 512, "044c53be1a24137e40c0c927660cde55f129d4cd8299f9799368e9f3f87df86d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "74d0e74a7f575632503daf96d966f1453a953efa4d2d2c8c4c27bcb6f4cc58c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "75cc33961899f5f0ab21606a3090be82160d09f1c9e5886c966182fae24f73d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "e667f1e0e9239ef1d13c1eacb7b8f6fb1a7e2d8d19f1b37227609ee976b48f14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "9209284691a207c20e7ccfb8e62db846361f18e9b702b3d14dab1d813765345c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "470a27804a774b83d45665937fbb9ae944fdf909b686a2e0ae4383887e1d2eca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "4417bc692d0c4c4853d116ccd1859b69536e5189fbcf2f677bc3a2212bce693c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_geGlu_dynB_sm100f", 640, "63f5bfdc9fa85cabab6b92b477018e5e5acd60f648b59ca5e6faecb7a3b739b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "156604d484782943a8aabe0f80c69f7a38afc5f624578882fb68669138105df1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "1443888aa0cf08fa9c99dd7ee4ae014f4faea2c7a9e30f0b7695d21bc1ad3d49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "abc8c8a08126ae761dd65100f1ed4825ab96555146aab36ca67b426bf1c11a4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, "050dd36fca68c41f82291b3ae65278b4fe70bacde72c5406e5805246e9da7b48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "b1fe391d871c6a74fa1471545e9a43092883aa356079e3001be370499e12ca16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "88282122ef2a311217bf041e7f88d525604a3c61e3a4dcb5905ab908a84896a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b4d1c479c5ff73073d1d33fbc6173973ffd354b228743b38fc5ae9269145590f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, "6fa168adeb46d156b427dc64e2b67b1f2c59b7c734f30537adea0c82264a364e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "427ccf38ac421d8fd426fe424bd32b00f1cf6ab30330c872ad913c85ea103d8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, 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, "ad5b8d3f5207618b0382af53a62b69664ced659970254b421b935cb445d987a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "fd0cae90ecd30bd101e0c1dcb0776e9e924c191b0c059fac43a24167d60a73a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, 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, "6291a688bc87c587e837632bf3b98fa58b40bbf5d5391e79395ed25366870d75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, 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, "a4f968a7010cacc14d3724675e368d10b8fe4e16f20e9b50951c4611021f886d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, 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, "d7f38394a4a5858051aa3970ab88c769a99de3c1b2fa70ee0f91b85d821a297e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "da9bace3dfce8009efcbdb193137f83dda2e3b44d0fb6f0df57cd74bd9d883b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "195b863c6fbb7b13490ffcf160c86b37beb451d1f82af8117c338beabb18f104", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d24f597e1a4895f9361994619340797e52833c707bef2651571c4222351a015b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, "c85594e9231083b7c0969903c3199b3fe31c959450d1b0d824e7e5b543a18b3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, "91b0c94b3750eab275aeb90acbcbe42f1a212608a7584625e03d0dd9a2f9a28f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "53d488827e0caed7c01f0aeb87ecbc03e34a55341c692d634921840ac8e4ddf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e663b46baa3a4cd30bd651d9ac5373eecde69772540ab52f458bc4e314e66790", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5cbe22e87673ed8a242b5c9cb6ed34110e11ddd2976e2d323d97dc02b65fd74b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "7a10bd43507f0793e3c3affe88950b7a337d93030508f9db0e8b1b9908642fe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "a721556d11bb7ddd17f295e3dab7f9b6f0b32b1a1c218bf3b887779515b6c632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "00dcb672838ec7388ef20a99a25de2b83d9f2c95cf2c3d08cab222ee491cc324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "6de88f4aff3ca037164621a1b6763da273f19174db1721c5a0821f8090ce35d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, 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, "ac8810dc8aa7e5f5f34084394ed5b25861ed13898f1c06dea20e8b83add40723", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "cbfa05d2d857ca77ce48ff3945d3e83ee9494c168b972db798a72943eb64098e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "2871935719190a3af193a9e4e82431e2910f4db5de5e08027d13bebc1dda60de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d9e1720b74f1d6bee4f3c9409d68f1ddc8e68ceb3d7a2083e0530c5b1b62f7e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8dd34b984cea61fe618af8dfb18f5d51e9e194cddc419b2337a8302996d2346d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "aa1f8077373eb8a82ab728feeaadedd0d38349576b8bf1e581a8da3e2278d08f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "4d9786e789415ac77ed1c3e68e656dcb26e6225a3c46f294c6a8c8aba9f83327", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d066cbcad305fb57e755c8de6a904e7b0068e5437a022b26ab043b96dcfde67d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "ec4c53b87eacd6ed8bdf041fff29b346a9c9512d5acdf61488fa4200ece873cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "2960cbbf670533eed015715e0c81a4671c750293db7851a7d2fd5ccaebeb4ee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c9a638af801360706c33f34aa706f57fed31524ccb603b9d6c794c17af6e09b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "052e9d1ecb10aa214ac250909a4ac15b82d7cf0fae3586a910d99bf45f85c4b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "31443b20bc4542f8a0cf02fdd55ed57606c00c353d8080c121ee6fb9363e89f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "829ee97d0db15310581c6b579b15b1fb841a733b723e4c49d59f5751c5cf77a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "16a75db5a8e123662b21c38c4cc623cbc8bb1b60ab21cfd080a7f6cdb4733a08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "9dcb220c5895a8432e34a212c1deb83a2eead978ca383234ecffb9030bf8a6c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "47a880fb000eea5f216d4c83a084263c5887c66b367c7cca6d2bf3b6791f224f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c782fb4642e06292af692ee73959451e4a324513d41f38dfad8a1bc6cffd06cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "201e7c95a3558382a67a38708291ae47cc65dda70bea14ee9e73a3d8e897dc60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "10fb131f73476ce776ab7383a954cd157a2a2ec8056a971f054a9c3b21c21944", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "3ded3f4326afcba629cf86c5ebd9139016a98066b6e4e0a39746894979192750", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43ae1ebd76259d35f444d728e4505b71d3453ecde7b03ef002ba3b25aa907c8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "6d98d0d6c3e5adde8c06cc52631381bca1897b627644e97f5fcfa515079e001a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "cd154c85593a3d993dce5cf2a974725ea67efe6ceb011ff573256049ecca74d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "94a293917c793b5721deeb0822f07d849debde79f6cad7ffb3372ed80492aa25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "746f091a9e529f1372b70cbb7702a9c5122f11c02a4cef7b7a744f71c447e575", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53762dcbd6f2341ea76132fd28374d66d7248e98403b25e1cf10f66633714bc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "cd9df9fcac9ff9040a85d7dc60862b43515b6e135cb1c3670edd602c6f00227f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "93a6d7ed34825de21e1d2641ded1d754bb80030c0a951e9532b4bc9c4608cfc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 640, "d22708bd55d29c21dde1a6c66af2539e0857c84ce765c4c0f9d36d4c5fd02d99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d903ae1b55c0d9001c14be7fdf8da457aa3c7f17af138664e34fbdd48676109d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9bfd720e9b69c012cac0c2c52a18a17a48979bedd53207735761d9f6b544a1e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "e7c57ae5f8c9b3d1adb1a0346e6fedeb57c738744b82a8fedd38e88199aca5c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f9327c67f7fd8e7032c51f37d0a6125b2dd3ba8d815c93d6d5ecc69784e665d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "cc27ce78c7c0ca11966d53b6d8905c633b8c2b3de6a92eb43f02a811b2fa04fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0b8ac5afb71bf59fb4dd12c8a11aaa2fb26832ce6cd8180b83ed178e202b5ca3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fd6d56d4af0c16e9ef61e5ee6e04cefea5dd3d4214ed1bb1bc7859ab5bfea0cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "905c809eb9e5bb7126e1a0d9f5d3eb176e010d51dae694980126580b4d6dd5a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "fefe24052f79cbdcd6564291d260fe631f9b33b4558bfa100fb6d528a165ab37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "36965b1d6ce41b758cab33ba3b946610009b74ae156110ecf72b04acb457790f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "590a94b917ae71a6bc5c90d06e0804243dc1e7b862198475004d379874a5c270", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "191a38e35f4ef732a0987dbf1e348d60c69cb7100bed6b37972dbc14b9da7b31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "486e173f33082e220c3deb2700f417aeb513df911f208ffafb9f49cd66523651", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5928522d26fa9c371e9560b147ee6466a78cde265ad70b4bb7c4221050d6ea1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c980eea06d3dc9406394431bf86c8c3411a4e64e0462811563625aeb38c81bc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "82fc4f9b1dc4829038aa7f2e550da9aff948d997e85b8b40361f7b2284fc255e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_geGlu_dynB_sm100f", 640, "a2eaa8b9f91f36680177e9519d25ca80c6ae6d37709de4dc523689f4e2a1d71b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d47ad9fcbf71f895f5472388e2d165bdf64de65e16334b2d47e2c30e51339ee4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f7f496583029524624f578fe409edbe0b30b481a9f506b4a1ac54344d9d1845d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0dd153468012c4d07fb54473499698b524f528e9e79edea9ededfacd072ec462", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "570b729ab3b793cd47d64172724eee501323204084a7190c5a088147042e7a75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "cfde476653708b600452c462deaaa6102ff88ca45f37f50951d0d9250cf18dc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "dbc7fa005503c346201737ffdc7807bf95f274cf6412c6617097277deea5fd3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_geGlu_dynB_sm100f", 640, "b903120d3fd228c5390839c6f16f22b433a230fbcb81e91e900ed008832bd1a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "03c6821e4a009adb080cdbc1858b49e3df11a399d61058d027fe38addc2be0ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5c8c69c09e530efe09326fb21b73c82c4e4c7282cc76be12681a44641b83f34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0fab515571c3be79ed9db17af38cdd4dc3c933dbdd29bb54dc6f0fb379f99127", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "48170d9efecd9017b24f2932b727f596ef5f59b66a728cdc3caadeca61211781", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "256bd7ee40e046b72b7596b37f34a7f11ede258843cc05eb5806e15de4b0ca2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0bea710e9c65c6457799dd7724869b5d2fec4acd25745a17efb546b16c080250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "ea012c8c569c38aa380f0e0b12468f6415e49a0b457795181ce6e829b1f1af84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ffb4cd7481247ea1c982ae7db51dd134be71fc9988432b8a5145242a9f8f4b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6fee7626fe274f1fb64c5fba9ba0e3990289f1766c12b6578b4bf3655dbf69fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f083a476021a0b386c41d544bb75147afd69e86298a4914e0a0e479fbd909eac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "ff65ee1eb9467ed29c6c185ec8db09d9b78ea14d3586693f46549391cb0c9860", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "078f205c849539d188a2d3d7b9dd56c2f24be5f8633586adc52105fd78c3110e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b3702af362d1a90ef369d5feacef85ce2411e803bf577d0515a28f4c8a619830", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0f4d02603a901cbf145fac48004ec6df2345c8871d00c92b6ca59d82fc97baad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "268b64c2eca888582fd93e0fcdef6498199f471f69569aae524be96d1448da07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "02947fb5d5f59a57c972fe9320f3eb7213e85891773642481131a3102e7ab2de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "db5fcc56ad6bdd0a3eb7de7e7d7ece1bd729a4507ad0b53a9fd59f532d7adacd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "516a12a29aab5566344717273c502f88a7cdec7ab1d15a00306a827a81a27338", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "cb62f42b0852f79a8212f03d8b5fa95b31cb8fb0c129d51cf60cbe81b74b4d17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "92f982bf5ac6addb406f1bf1ab20477af09ecf4a01f0b3b245fbc165baeff659", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e15a41173e22703f6ca5b480a18454c6ba47f3b0909b555eb82b99fdd1f4330c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f43bbfd12a864d105a67e94c02330017753b7530a51dfe003b87a9acc43411f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "9a59825365321cbda577266a37b8832016d9bb6d57e3a79f9d4491aa4bf7dc20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "43877775ba777a9d420703e1d2b5c1c21e00929e06998b17e820cf72cb7dd423", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c04b7d1ad12ad4ac1c42a5e49ccfa9a18d95aa7e6126e449f571e04c48308b01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "c993487424682e5834b4d976e3cdbea7b5bd10cc8f4b37c13b93f7375b03196f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b5be98432ac4ff2a7f08ccba4c0076ea100f583ab2d9aa44db5a3d6f026fe230", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "a6aa474bea695872b02df19a1b05806692bc620532275889cacd54c110086d37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "06cab004c2bb578bf04911892aa19454022bf9458605a28fdae575e46f715f74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "5df9a50feade876cb1158f66bb6f99f4d70184840fd5b0c9754ee49c25b987b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "2b7d12f134ffec1981bfceb205c402cfcd5d2df6a6a9e7cef8ce31ab3c40b02d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "7345ea4f7e00dc7fd0a7949b30123308d7edb98cf4075e1635a1ad963af2e5d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b8cc3949ae3ecbb55b174d44ea80c6d2b75691e54b8ec6b238968b54f707e44a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "aa92d80440df84743ed94bd186efe6d6f70b9986158f75839c2178b3ea5c5da1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "4b23598a8defa988a0f3fea9d87d75d5b9fbc811a55d4186e32a1bd6b7e0a040", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9f7db27dd592233fba9e2ba53336da00b07f46c2b06dad94bf8a0a758de85220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "4c3a8fd8caf23eacd6eeae14a800b85331c3b78bd322a15c0ab7fb6ac76a48f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "1bcd71050916ccb3f196ac794e660c6fa7d8ef100dffd742a922c744adcb5073", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "984d03d0c80f307423b16aa2933304162d8a75f3e1872e08b1bda2a89910c87f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "4d85a5152e3496d1b1293c3f808b1e81c20cf5cf2716cf43b916017bb0885a98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f65ea9f27d268f6a9692b9538426782cd6eec9103be504f195bf8bbae38452ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 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, "cc36666b4993dc72ce882cbaecae7f76b0faae63208d9614da98760261e11715", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "cb4af86e1d84453ba05143508b36c6d074d3e2f12d030a9c48d8030eb5891025", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "13e51f7334304f92707e3874e7bda2c5dfaa6b3ce244fd445cb7ab5e2afd30a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "a1769784539355a7b4c3bec59fd289794b56138c7a91068798bcb6b792d23267", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "8a77c45d953f1e06dfcb76cf0e313d24fe2b4380165ef5c183f1bad2a9554c10", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "ac2121e921b0cc1acbfd1af224c8f2424a9445b41399161b1516348fa4c833d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "474ca1cdcee388bf61c9f56be710737546e283d5306bd53ff0f9955a04f44551", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "fb3ed0a78d9fa6807a294f65a237120f814ea0a108b43e50f1d3079f8003f588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "b55d29740deabefe048b98f22e7cb636b41860d504d5a1a7ee0a3c773f14a71c", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0705aa4c19496e1f5e64f40544a14bb96c72ffca4b09cc8a59191135eefacf8d", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8808f01c20fbe88954bb1bb2d669909c1ac588403b42253a3d307b123064113d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e8d3e903d09ef9e28edf5fc5457d741d8321316c9a22d0b6b93b1bb8f0b75606", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "744f5958d0c7f537ec271715af6726d0ca325b261b4ffe63a69669c6bae2905a", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f832e330ef07fd4d90acaf888fbf47ab71d6c89955af1649b1e92b2d2ce46b0f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a07d32167cae8cf11e2902081bff1aaa21607f011b83b00ff44753d12b04d90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ca8d5a4361e6ac419b2ddbc5d67d9b67ac511289dca786c00cd0c336caf27db6", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "75a76809db58999fb3020a459284b6ddd2c766c18d74dfd2a7a5664b9d0d8f42", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "746a13202dc95057a0c5ea3b40896f6638d65956665024d4ff17a662c89055bb", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "0743c1b8411b7f3471f2caca4dcfeeef001f7037cc95ed5c86112681b5454f19", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "92b120a48f2f583540e2b3e7b8d2276642953c3028203b382d17741fa8e5cfb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "c7aa6399223ec5caf804a8cc90f66b3572262693a087cf99066522a3961d6eb9", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9196e93d6c35246481b38608d90e9c6582d001b1069357a6fa24896e00e36280", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8fa53ffa35aa6b5678452e97f01ded0e7980cc0bdcc29f5aae370de219192ec1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a3bf326486ce44b82aa5d59190550c1d31ad2c78dcaf822109ac37221bafe39a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_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, "6c5e7a5e75c95f029217863765a86b219a5bff62b7c4b4ee0c312c176ad5571f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "d770e96a79c2eb3fe377e32936b8cc75254942422467875094cb196700485c8c", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e3c10a36483d30b0410477b85185db8b05b6249d50047e15b174b89624107301", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "67733b9f4792afca4044778fc90eb96416a57b1c02bd06171bfb6c1cbea34333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "390e179124cf44d5984ff352ea0374680e1a85a337a1ee124a2b987f720810fa", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a51d0d1094969702d5f00c808e8c3a682c154b6e069de23ca63869e99de5afa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "35fef80677bbbb824354407b8a37b74808bd6e8d51e87756a0fada5f0fd6bc90", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "299947a98a9052040f306dadda0ff0473e29281b6921086640c0cbb0043c8b2b", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "9a69b313ffffa861500a24daec218125dfa56ee17a80db160e8ec40eb349b0b1", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "33ed510004ada905d1d0aa9dd8ea2321c8d5e4257c37a2d3198539057696e85f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "56b004ddd66ae9d9c78816244ca1cfcf93d42c87a359026749b5dd89f4da90cc", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "7fa9729eb4412449314dc1b301fa8d966b0a458104cd93f55f15b548947f0259", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "82360df388f72a721295526fb9118554040eb0eb9cba1715780cacdcf5b24b9c", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5122e3b4a21dce8891447749fa4caf59f29abc8ac5fd6c69e6cee72259ac837c", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "24838fa05d4939af812d646636001f08e699911d7828441efb39dd618c196e42", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ad4873d966a5724bb0548bcb500d0703ecf1fab648df39e2af6dbb9d45718d5c", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "744f02d1523111a1e47fac0b4641885046be1cd1f6043f5f0a09fd2f23ecfb77", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9521cb22d4f7414608d0771ce22194c99516df769e24967804a02471b4c8235f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "e4adc0ee386958f010ce9574de9c4b38a4f5266574a4d6dd4b72ea3ede7ceae2", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c2a3d5778b4e7b8780dc1534117564cd40b28498d77bfdfc28d679c07c23fa3c", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "24cc140dbd17e1c539f55278bd3f06e8a5b9092d1d63c541bb127a8b11dce195", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "c5fac047d316a331977542a00b547d1c7a85c56509dff26f428626702001ab33", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9093aaa868b1b3c984cced31ddae3989f62ddfca51ccb3e8080ac28279634b70", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d540d3f035a4831e2f365f8b13080a3b3c7fbacb0841bb45ee2982480456d653", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "392ae61c5c8910fab24845cdc68f30fb0b4eefd92de51c56ae08c415cda5ebe9", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3e9c51e678b43b897aa71200661072ccb5419305cfcd77e915eae0b8dd9ff05b", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "711859c4e22677adf1808568c9517b3bde49b39e9b462f7bea949178e7546b25", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "11eaacbd0fcda242dffe1e0d131d812132cfcc688747fc2cabfef092d2980759", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "41faa0663e6342d89d4aa11d312c4a47f37b53e6c8102b7743d2eed845a474c6", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "545d142ed2bdee53982d1e4a78505b49d2a9f859f1e07591c72144ffe1406b85", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "9081a5eaddf9dc21ffba44c7f93607912ccbebecaddd5bbe7aaa610926d705a6", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5c0369994f93a86fb2d91bc1509712cf4840931b24885647e9deeb6a990a8e88", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b77cd5be17a7a589dc4f24db20a3c61291ca011518f709ce8d9e3b14217a18ab", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "16cb4f4f423e2e944e695df19461723b12e9459bc101ddf5b333a5861e0ac6ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "276e7600cceb34038fdd9b7a51bf2f52065b85412aba5fd0a606957b04de558f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2b76a5db090203fd7387a1a1309051b4282fed6a900897ec5230e9881c1abf5a", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b9f41429c169effe60c7b9ade9f0b72bb2564a1f6a05b0324d0d87e9e836e712", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a5e80d7db13a62d211a073f7cb847c07ac90d96363846f58bbea5c88a4214b47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0d55622b1921168e9d5f86320a0e0deb294365a36570ccab6528d102f322903f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "88a7212b1d203c20d7f2a3a8ac970e879834e0b2598a63d6be603043983ca1bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "49201d0795fd2f0786e92dc5bb67d255c2280f36af5f8b4bfef709d504e3c7ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a5e96b5529bc60d8f650951070b62d3ac00fda93ef0ad94283b0f7e556bee815", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "9c5e846bb88bc86cb96ef2d19d5fc0179a7b421ef02249de2d98798bbd04e714", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "214316a95f8b2a43bcc740dd94844c562b942e9f8d78ba1b066e21acd8ee47bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ab531bd75b2fcacc6e45f5c0ef7146c96ee87da5ba49a9e7f8029017a97cffe6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "781a76d1a7eb381711a2b2167e422bb6c6dbc797d576ae64e6553acacd9fac88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "1b573b04234da3a6de70a4cf8f3b4c69b2c695093978be1cde52777072888516", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e3de08586f872f4eec772915189fd3f4fe3d68a2593b55022348cd6d5c13ed96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3744c837fc015997d9ba2808f80195a3e5f122bc12320d50f78d2fda75ff5fea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "313b2ecb2883f1302a19d98697eb2955e0407889c0b38b87dfc5e00e81683b30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8549fd7fbaa2ef5238993598a679f0f664e0df0b1277d3766f43297c57dfd36b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_geGlu_dynB_sm100f", 896, "d1c99036b2899d444e9671dd95953deda67ebf5ec858b1bc3f8f59f0ba76a835", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mSliceK */ 0
, /* 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_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "450826e59dc6287729903919a15a4a74fa51bade7ae5a1db41c55f667c89cc14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0266a8222f6104dcd9d66cc6e869f2b2bf2a1fc188ce4e96bb35153e042c4c84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "000879b1aa66c69ca68a40f67be5c2ed2073642b2956fddb335c2bda000bc837", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "afcedfb6e84e28496bc8fe393fece452c3c7e7ec743a22784aba86a5c83003aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "23bcc384f3b3216fa4981c9d0731d522d4d40c532bcf2b79012a89ffbac30065", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36bc8b8c63d170b48216217fafc21703360264595f188a923140f5df4cca8a2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f552e887db84b0e2e15002b3b27ea5fe9a2e242f0e65bcd2314b28de82addacc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "026b9fd21f59007e70d5bea31d5bce0f8bceeb5a5243efb7c7709c85e7f1febf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bc1c3281e75187a7cada843a6fafa0f8766f7ca27f313008490c29bec2d48a37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5aa7c2b37ea9a07263a1a77f2b7e24f6243b4bb972e13e36c130cfde97fd5e9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fe2b8757bbeeb7d0cb597fe0f9843fa3e4c38c055be2dd951d737b794cecbe35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42ae933eac6226e8d7220a0e7ee96bb2e00c42edd746365721db81f7d3f764c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "510bb9d5f0cb6ebfb4d57068a3c57db2e95b3b148b1c02022a1f8a33930f83a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95ab8fa48fc0c3b89593a0caf63050a09470b7f8f2da101ea016cf489d04545c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f58fb8c9e40a84fe288f2499d9f13b39abcb8afef42ea07698a8a89fb85bc99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8ffcd0ae2c72fd0a6a8814503c169110eaf49796911b2c04075832919a018b82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b8032a02943b7b53bd2594ace5c0f60c3fb27c5914d4c4a531401a0398e2cc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "c650dac611049278f424c845dead3f0b9b0fb02bad2bb168ebb1afd75072d7bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e48d8db2154fad46b8014a13bf6e4dcdf812473929e8383d9bfa01cac686220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6ddc46c948a1e27eba8e7f7e21db9df9af6ffcfb594f4cf311f2ba07f9203fee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "fbd079adef600b4bf22c402272293caf69bf44ba5ce67eeacca1d27523311c74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0a4a2b6343e97411e485a3cc4240682c25045520179833249d9a3373f2325ed1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a27acb5321be233d69f4f3eabd17e41a4c62856b181d9884a097a09617123e07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a0f35b8ae0f3138f48b06e356a7c329c3a495f04ae5938b6e37fc9489449b03c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "88c582018719367965dab719b5db63c0e6abf78c58c2cb7cc98ea9988631c8de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "39e63561249557b0a5315c0a7a0be5314018670ce60d15358496031a505bb3a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "943d43e988fc91fb84e3eed04d69e1d49d26be4297a22b8b1ad8ecf7457569c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9f30c3a353abb6b4074f9c6ee9ae324915e46d6a67bb6ff783243efbf668a5ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a0db2f37a1a7a04fcc8f668aec818c14a877c73db557470b07d526f88bed4e5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f6e82a0754805ea0bd9f820b8df2b38f9c25502e62a833f21e96b2d03738f162", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f7e5c0a0b69244c287517bf4dc101389fd124a19f63423dc9b22605b8cb4eab3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7f220accde1cce21ab29a6748ddcb437ac6812d9ecfab7bc0521066c53b15de7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d231a259f09e5758034bc55108995c3aa574f63a0c7973cbbd4b6d094a90c87f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "f3d00191f6e230e97a7c970ea0c827e516daa52d0e1387db44a42a45bde8ba42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b2c6e3ff553457f78360e9e1436fd12caa87525e9f663d0ee09a86d68424732c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d4df91347b49cad40de3f371c04d4eac5127eee407aba7c9c9334c5f67e29b4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "6fe579100f7ad3773dc963e5b5f98956f10bc7e0aa47eaf0fbe813584c2ec6ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "058c330a8c49fae1ee42447bae38dde370309da6a2c73cca56285a38259cd985", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "ed623701aa7177c625b4b1a53c4ef5297838565e627f7dbb9646fa51e73ab642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "71686a959148ed61f8d891b4e38312faefc17b62bdb2dc08ec9262fbafe16ff4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "daa92f25298be8a1ab9485f6f58e593e545beec178895be07b30230a14d91dd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "97343ddf1b7c3090e0390d712f82044e5a2c8c3162ec56162b0d17033f9c0fb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "30a4ffb5667ea6cf76930b4f4685cfc7a7c5b7dd589cdaed1c59f85d5af28625", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "7f79ba26d46c7f71495e1e7abfaf4980ca85f9313100ded4c594967d7b5ca47a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "80a1716a7f73a64fc762f69ced049192603683354896bfe2adec6b0a14e87f0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "46d1b15a8367c08682ae9c6e3fdab304a4190ee7021e91791ad1338d0911c139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f6af2ee804421100f2f4ff6afa98593dff4c5a63701120208eb1b8bb1a842575", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e17ba8758a54cbfd80819c05cc230c850e43e56a882ee20d06cf08f494c91c55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a3d4a52ccf1a037427cfaa322ceee924d74ce1ed791ef83d47ffe1873f73d188", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_swiGlu_dynB_sm100f", 896, "6fa644be24b683a2b459545324813838ba4a8de1e4d870544d7cd946bde0172b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "40ccf8fff08341a397d202a8bafc1d4da8648cccb3eab7423f4a7c9e8c055ccd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "81f3e54fbba67fc65cda46f534ab55d0759f3a2c4be7647907c8f8366ecec19b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "badee3584a493100902a134a181936b7c4cffbbed49bbccd0cfd173ca11ebd09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "85afa0490a0fa329f38a2bce0a149621ca984894bd29552c38981311f705c0b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "3575db6a329bb329244c63387ab243a94214b80b3c69f0c5f1f41f2a89819050", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "1c88dd68e398ffab2911fb86757ab4127a2b8363f05beb55f543a115299d70fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "236b8968c417d279143bcb248296265806996839c84799ac0250fb20fe7129f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9b19b0c9a6b205749c6a2f516531ba21997937b0c92c80e08ea0945bbf897a3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "cb45a496b0b9bf7f4a396463e9d25593396234f2f807a9785f15c69b525ca3c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6c9d60cd92fbcdd3834203de54b443a937305ab568a8983ab7129f6f448f8b65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0d8f9f3d6dec949e024292d1045517ffde588198c1b06e62c22261661ebc195c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "ed1a21ad01d5b55ce124b335773a33413d008fdbe5368a8aab1c8a5a4b64582f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f26c5f74721896bc04aa3cb20085330d2a79ad6da7f5b0b40666091f6b6650fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "8d77d8360643a68b4d408004885cbd317bc5ebdec6e2d2e961ad2ed3780a3476", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3035d2f5aa4071e85480d8220aa6acfe2c96549fd2573f36adedd84d62172431", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "bbfd957121e201657df98b11bc787a7ea655bffc417b8548ad8f7f4720953c33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "8273abe8f38d5db3c0c117fc6568d93759e93d3285c970520b7e0cab948a8802", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "9f1660284f522044c6474a0f727e31e4fa4ec817f7c2f955a992153fc463c75a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "eb07298a51d9d44a38e09240bd14c5d1ef41390a9c0e0d99fa98e78c3bcfb640", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "72198e264b65d6c5f681411edb24f1a4c81840a63d9235f5d24045d87e1a9293", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "190e1085840fd765818ef0cbed6ccd92148eb187ad4f65e1b1dc6b368bbfa118", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 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, "accb338c24fb6cd3afcd0b737a4a1ec8d69a9dd9267bedf39ac9a397f407e000", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5ace93e33f67297b700f0fce00c5c5f23cc6af09c6f067af5baa3825c1065aa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "dac3d2eebafd8774fb0e551a87ecadb9918e0b9ee94e93346eb321f61640a458", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "37b020975d542714db19f2829e1408c8ced3d80f9118cc44813c987c5d2946c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "6542ffca98d0f7bf7fa3a6223723597042834598fe0a4174dec53fb910ba233f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "f91145be388445eb644554a9c98cfa4aff0303aaebf6cb243e9c922e8cb2bc12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "22ac8f42ca27f8fb6198da1b4e6cff6f4caaad1c4f003cea27a49fada506ebb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "eb5e23bd28b3c7775005a45c2a3b4b36e33f2a818d4c76e19d05c75d5a264a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "a5a08681674b43eacf78716976d99a5791cdd7e7bd94caac0a46f1b715d9f5c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "c00fb215d598abd6a4e7368a8b0bd3f9036be588e5c547da6388848008d00421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "434e1094990e6bb377c21b648ef6fc359fcf3f4a97d29c7324f958a82888c7c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "eddff5231731280e450fe62a2d220afea0d2813626a28a6906fdcc4b4944b18b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7498e0278ba9de0962fff889739816726f5a794454e015e60a083999e5ef959a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "8bcfeb587c5ff96f13c986b3065719819c4fedbf9b168deeb2b478b7825085c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "11b642643ded65e295b8a7a97e17ef79f62a953e4993d526c53f5b929541d57a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "950f4b557dae2e939c5af884e10749e998f6dc20f9ddd56cc02cc006f5fcf4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "f3723ce72bdb2c5d454b7a2be8bdbe0ae0bf3295e5e5a1b4a301ddc4d6ca3360", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "1faabb412fe748330838119fc590c7092de9086523d681932949915d274562f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "9ec0e2e9cb3d3b283edd2e5a49e147867bd5235bf3c4ea3c3b9753aba74e295a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9599001a7b9a50fdbb1dd72a073256211111ebcbac49a9d5df3c08c1b943c95d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "897204d4abba47e61727ee5d6b721e060c01700b99a6f6d1f4fbdfc2e1968f97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "72e0ca44ef7d5b3780fea5ceb1cde88ae43b411a9e55ca48505576e9db3f5a48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "fe5a87526c48ceb004a1c5c98d4ce67ffd9c69f1ab707995f4e8f52e74c48592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "eae5fd60d0650d70dca0545c2e77029df56c4e99dc4e82fe31357f32682c18db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "bb64d9ce5f8e4b8b3a2d1f54b71f2c157347b112c94051975fe1aed44c376154", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "cda324a1d56918f5afcf08fff53e6cc9b610767841217d1b2c0ed40ebd168794", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "903e88340267d120eee2c92e2b915ed11d6bf400b51538537cc624e37018137e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3c3200f612f15e9a14b730fdd29f1867fa44c47031c86aad4f3085d0bc1b366c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "3a5ba47fb7c2ba0935a4dacbef47cc716b8b4ff115064e8932bd6d977698fd0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 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", 768, "df61686a81cbdbdf680a3edeabb4f9695ab69727fe08d89fb53243422e2cef4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c70b74ae7fd0b39d1e28fc887c0f84fa27d5ce619b3128d86d07bede966d2297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "cbb723b4b54933a4bd83e3abc8b0cb73bf3231f7287aa022d8affb29efd2b23b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "194756a0762196c93a698f97a0a3af968dacc965d2e41829aeb10e6922b856b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "99c3efe2ae8408344130235214e92d7ff607485a78f93708a9d78c10df6a7bc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5cd62f6c24e085e8810c725e5756f58b6dd48080738092708aa796385d3b0cb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6ab432ce144915b9c3bc785cc1857ba5eb99d446f6b8468a469ca552fdfd1f19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c63e7ced5d8d2178722a94e31581479af68d08964dd13dc34e6ef3469456e416", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a209f7df7557be5eb70113b6c1469a7b8de085bf4f1c8211e50812606e519dde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "039e4478b3cc0b686e9c1cf8d8663d08c6a935a60cb8192a8a017612ccd9997b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "13a9afbc3cb6a4fc1ba6f522efdd9dbcd1c5aac10a5efe2cb3da19018c90facb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82107064a1e398b11643235d44b18c6dc522bf69615d020f9b5b953e081d6fa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "49c4bc2d638b70931d4d843004919640d45712fec86196234647c257b8061566", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "45413b676a663d9117aa680169aa264068d401763a9c2f96033d8f13232b5aa0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "eda8a5a5882ce188f4b8579b6a9184a266613f31388c830e0cba2c2df98addbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "03170129892f965cdbdaf06bf280280cd24bd00b0d75435c76259a1b8178f58e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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", 768, "cbad59332a1af353151d468b5a5258bb0d41a84a835e5ec49213d423bca47b4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "968a489d3cbd5293a2bc26a08cc9bf20a76b56cd6523cb36cb79ece5163f7858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d3709189002e56a1bab6ab7a09253ca0907d7737f22cfc6dc794ef9838867c3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "eb6a5b78e29d9cdc945c7204ffe3ce3d5266ab9761e3d411cf76936dae1e7f6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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", 768, "c47ed8b9be4a8d8be6fb9d5989993a5d31b7f47fcae39849537493e605f4df4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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", 768, "818fac91d978e46390417126ba94d2132d891522afc445aaa7ccdc2d48795f60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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 */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9570a87959aef88dd9b03928af91578047452051c0f2f3aeeb522b9dcbf14adc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "30595c038ea4241df8f6cfd12788df37050f4f019683f7567a2c420927f43fa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e2fcb8010526bafc91d64064a3047576c47ab247e3f7d90ae0ff0664de09069c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "bee7b55a0b77145396162e7a704183d064934358b015e16848a6310bb61aff36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3c7436a5b8d5ec13a85743427e22327f5480148f6e18505b00ac57eef3bd604d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ea49cd0c5daa8daa89770ee3fc8570f4a2836b444bbe32e9514d7280fd8b819d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d01238c56334ac28370d2fc04e283d9ee6d0bf7d79a157ac3f38898886b4a86e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "d45d2c35446dc073744764c1322106d51e0d71ee2c32eea166fe042d4cea2744", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8976127a91037397f23308a8e6f8794867c28674aeeaabe3149de53e8b336058", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "d546baf199165562dcef4c26da815380fa05059f3a08341910d80a825642f183", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f348b9c4d75e9ce51ddd061b6b4058ebf68319f555c04b66dcfcb03a4e849e12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "61a69f4099d47b52a7f9d730f7e4a0831f206719c6f9a74c059020cc7dd9f4fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "afa767483ce666cc6c0ed98e5c7b18e884e1fcf38d998b90edf8c8aba8223595", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "169df7be270bf8c293445fdf6af0cb1505380d7c7c28fea58d83375d13b5614d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0e9e02afaea68499e85407eb790a0598fd8e46b49b4e543125614ec8cb153878", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ca53992b4562af91492bd2161d8aa9caefe711bd2236a1cbaabf3fb610914a6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0e28fa95878d73353d77ed5837c15eb5360b29ba36856ba9ad48223fa5c51989", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ad54f7748dc3801dad22377ef368958348dd3546388d314460304508c99cd88c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "bc0ec1ee07636e655fd88915020c8aaa82156ff834a0433a13e9c3f89eaccd15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6ad6978f8cf2d53d850691977e925ef31d7bccd3b901cf66a8a7a0a06cc95b28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a67ef4face0e4ebfc46d456ab28827e530c1afa525b6908463de8c184966bd7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1c3002f4ef47b837e6a2bc204c573212c89556153a4b19ed22e8f80271f7d28f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "401b43213a3e68d889c13af64300c11af75378c1e7e90a9682bf8c747568c5a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "10ddf365d1c0930e028c1674f4b8ba017ba7b7e94b7aab864a7f76c43ed8131e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "270fcb3812bd3ddb304d2b51a7afdf6482615e8ad9b33b830f08b12d9cd4333e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "119993fc5aa8af8998b71a99ee5ce8290a77baa181314e563e693514cca01dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c865ac6b37e2d53127e2177f117da245c5dc790f8046d09f6726aed5320ef770", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "91c0e434923192be05baa2fe488e3bd49e87a8e3dd97c56c5800264828d16be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7761e7298bb20eec924640ff114035748ed2e9be6d3576a9ebdec8ec4f74c28f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "05d962d9016492643a3c5e6a5d6f49671735c96649eaa3f95142ebe5197699bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c1e09c8b89f551dd5a025236ce51ec3a2869c9cab09b143817b2b0887e25126", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4ebbd44c21e7a1f5686a97ed96e9c948b31d7a882ab48256b79ca02fa5849f03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0a0fe7853a84fcf67ebda9bfa7fb826ed97daa439a4a80dda5ca43e348a96903", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d51ecb806012947e4cff98a0b9d273438c3f7e8ee3849c8aea7d207ec05a0bd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "21bf040a87fa55eb133c427f1d4e15f543943d810609fde8ee66f3fb12bfca14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f88d0de91f974393447acbb06423fc0f63163eec2a604186f4e885ede26f1dba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b36ee70d437d7bd86cf94aa41eecea7798a00f92a8554602732473fd9189f057", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "eb8a7ac0a22120c0374849499f817cafb33dfd802323dbd0d6a1e2c672be2250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cde8b914912a6f6c925c3d616a54ee9f5ff4042f247e2f4bcbc3b90887850ea1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "02b41c612c193394e8f6b2492d3f378d58337c03cc48b02cc6a6397ea96e28f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "e284e09fbcd9c6ece02e9f13752b29e2f52407fed04961f17b3140d9b661e3f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6f878f5b49da11cbf30a53d16cc8dc355811b83f02a065079454de93235d17be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f7c50c4003bd6970a91470541742c0e18a0db38bdc48660a094755d8f3cafbfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c47f0b97370079bd063634ab0ce142beef667fa3a2d9fdc293386aebbb9f5a09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fa954044cc8615cb35c2ee54d78c3314c41475b233f1140382df94109770c02f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "c7a84d10089bc82a24ca76a8226ab1745792d3a23ffdfdfb0f299fe9a951da22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c72dd1ec92a54cd1ae626041ec977ea6b64e962c86bd1d008799ea260f1316d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "700efc445a64875472191e5ef9753c2128172479c01997e33229e4abfd9a13fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ffda44c25395fa5ae6d2b5e8c95416cd6ffe96633f783e7276a7885507ecaf7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c15751481093579afb489fbd0f6c8596a1bdaf983370d95e2ea3a4199f0f4ed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "490ae984d222a0a76f782fe3b53b348a59a824880cb878258a4189c3493f0b90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "1970ed6b80976c16921575e1fe28df7e1a52d1f5c47f81a850c6c98bb6130dfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9c4e9af9b902ffa24f6e8f37c9828e7d9ca2527ed59fda4fb136f38828371f9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "f5530fb6db6336fdf9d1cae51186b4494e0105bac05c6fdfddded4c9406de5dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0e138f9aae17b7d3074b52999f33d13726377d56832c1af55496b0ad123e008d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a8ff1d6cec31e47490e9386e049eb5b811b678871d64a27680c6cca491fda1b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "37e5faf5c41257d72e3482b7622bec2bb44bd57616f5c0ac635a093fe4df8497", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3bb7f4751282d2acc5dd2bd94658b5d5c92e45e8f4aaae1ebed710777e3b3c5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4d8c70415d4f645954a21fad7c09f72a9bca8f3cc5eb973d5e7c1d7d98fed436", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "41f7cfbda79e51c027039c599708eb783a1c90a3c3ce40dcbc077f92b48b7abe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "889a5c7dcccedb96720bbff7a5e0f5f9ee70d514fd551108b5987aa7f1f0da2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "39aeb3acc8cca2b472be2c6e91f84087b550fb81bc7b7b023415d0c0e8889141", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cd9892acc458cfa4d404cd5e35506dc315e577a989e7576482f03be6c3b719fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "3c18b887e5199ed88917f49b61e3b73044e139952ecbbfabe7a427617255d7cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "24164c9e1a3424ae282235fa319e9614dff3c63f70edee7d3de80efa3c429ea7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "30868e4b617e36012f5e3b0b77c01ac3a64acd868fcf54851539008c8102e763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ed0f3f1c84e41aa60af2b6833ce92c35e20ba42ef814b89cbe9e05c66418b27e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "edc027970d92cc2c9b839bf14ee067bd1051146df18945e07f1fc424270aee97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2be817de6001ae52c736c03b1b2997f2bc0f40f838e99e4f547189c3f775e00d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3b601aabe31c4cc9f3dd08324527615dffd2b24cb7b6df1a393c63ab093f0783", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d5c2e32ddef693399574e88816ca449e5f2340dd67a8963f976c2e4bece41b1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "0f195687940aac815b4144f87885b41372b8afc311e1c58794f1b6d58c741b6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0cdc0d8f62953c4fc8e2896e174ed0d22b923e4ac7dabd07ac7920524389c4bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "b30f52bfa93fa968d754e1f6d7b32416cc38dec8aeb2176514ac2bcecde9b4ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a848b44dbafb99de19e2b4c5017fc6e3730fe6317fea8136c503954afa9d4a5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "4f6b4261e395ea9cf5565c8b8a50075a6fb3a7517878db330ea820ea8ab9fce9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5e965ffd86a6a6fbddca389c4e2dccf391c25f2bf331bbf628e3e96a279f751c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cd9b530a594e39a5556d6b8dd9a7ba208eebe3ab15d218aae4ba354339d676b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20ea1860a41bfe53865e4b5ff826f783f947c89ee04720f51888f2142c05ec45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b93a38f1ce7d2ac382617171d30db84eee5e9bf7a17cfca6952b00870f56431f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5360710a67633d8791bce51979dbe899951eaf40b3e588911b4755a091681ddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "76b1d64adc45a219b13dca6d83643c5437b26dfb97eb4de22b25d565cac79598", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2b221e9c5650efe61b83d569a275e1ba5e3d1b863b471ccdaab78b3fe1f67b6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6887894d274cd2e7d07d521bcc0bd93f12c6aa29e123b8594844a8e9239eb40a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "865761251626663c6a4653d8fada18748e58a6a7d7540295f0a6679baab460b7", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7e5502a1a9ad068737e7bd75fec27d9e552a1baf7a4147a7c685790a38b0784e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0250bb908c1028c372c55b60c474f785507060914e307cde5d8e247c0be1a1ff", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "747ae04a8aac99597ae6031902eda9f620721804f9d3c59e6ae8c97808fa8b8f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "44d5511d8c585b66e3eb5717582bdf242ee8e2b69589b6668e774ec7bbd759f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9d78b1f758fa1faed689c1bc20a6105ce2553e2bec61968e5255730c2f90300d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "71f879b88a930d1acba20528ffe0107cd2bdfcb3f75cb6ee6a3db025138d939a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "762d0d5e9eae0307dacdb360dc543cc166cf75da1c8543fb79b868260e20cfa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "5c9ab46472fab7b40f7ed4a60d54fed771fd11fbfe692390e9311ae6a477b52b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4e82e4ba9621c4ce386fb3459165bf550af65c142303e0ca3fb33e54d77a19ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "e3f658abc5826b9295cb5f34fc34b76266f484740a42f40601c5d04847adff6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f7e7aa96b24ee3272826e8f47d8b11266dfcc1e0020949dd8ed62d5126c61833", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "b9d3d49aec25df527ee43f766085fc08eb37e8a28ec3768e7f2cf7f1519e8a48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "90b94627b0af5e9d9e65801cb07c2abd5bae2f01c298cbe74d1dc604c622f5e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "b3d1ce0681aacc58a57e963f521f2c62813045fdcbafce7ed36ce19d90904edc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "200d3032f58307b7b103d69b3d94849afd6994fc2e716c65abdf358db9144d8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "290f03ead5c7480637090ffa9423908cee8e13b76ef6956bb2793d745d86012c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "bc08ae37330614771affff8961592733c456ff591e8c73df7ee3ebd8ca68e416", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "483dec2f4b9ac3002521d1bb3f0e06029c6c9168cab88ac67d278dcf562afeae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "9e31076058b32cae190ab5060f2ab07b767fb29c644ab0fd46e76c81715deea6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "eecece21d0268e4895be97c4e1e4a60db60d110fee9999d0f8338d36417a5db4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "186dcf30d03e4f3550e75be090e8543d1f26a7b7521c0110e8d9a5fd549e60e6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0b35467886ddf90111f1f6cdfa1aa72860dbcbc8411e36d077cdfc0ddab380af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "278956c390be632e02715334142abcbe6df6dc2b66715628ab5617eb15ca8a02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "02de543d879c3b1292e73fea317f453bd29442f98beeee5dff9801b67fc85b66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "e3286fce9beff2c6b5e0d68276dea7d2e15cc931244ac9a9b43e6509a80df04d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "57de1386b713b33d2784737e697193c23def9684eb17dee66752fe9d978af778", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "608fb3ef07dfa128bb7bce20aaf21a154cf87c76844628145e5e4fe1fe3b2104", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "423c17774412efa0eb0de232070c5cb6a2451dc34dae4bf1404478d8934c13c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "12433ac5b92d660d0fc448c55f3f08cc5ee35aa88efb2cc5890794b280c23617", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "17127cb59b50977187aed6028ad00023e860b74da822d31ed1a43848fc0b5f16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c601cc4a6549650cbefe65cf780f01a8e92fae10e868fa03193e08e53c4527f1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7216e06a2849421d1bc34b3290780d8ab850ad398be3e3de3aad5d44b7e3c2a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d7fbd2fc4514eed99cde1dee5b1fc81862a899db62946be48a7790a9da5fe984", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c451c08d3234861894bbfc97a4ced9273d3898d04fc5170337057775900f06ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7ba3004b725b0946ce7401f0bcaec7d245b22d95e2cbf3f3d7e26fb4d4bc52d2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "93130593a16b496205d69f0e692c7f1d628bb1acae5efe0ea6f6f822f1b9b34b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "903aa3eb15e7cbbe54660e6abe56504e6e1c68f774c65ae88bfc16321ebab216", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "09d6c475799e9922b03336b4f053f8150f72a7ceafdfb1fe08bab59b41ddd5c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "893d0acfb5603009e64cb081d8cfe31756eacfe8d4fe7a191755fbabd78e26b1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "4f635b611817c1faf879f568eef03845e4dfae057a2c4597bce0c5891b699f0e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "a3ca6b634de3f0a16df06523f4bba72f7f6f8db0ceda97b691681606761e06ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "0773053ed2263d4209331fd1639601a9e32b4d6d7ddda19eaa3859e43746b084", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0d0da62417d7293f88aa9eea458338df6bd958f0bb4bf2f7e18eb30a89308629", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "62f11a4bc4c2e6e9895cc3833424dd3881cc9509483af6c64876915de4f3d6c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "3bdd865c4153b6b1dc03f4d660da9f8a310dad71d80b32f285e0eff04f571750", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "62fc8dba592a608fcaac4176db85a014f6619fe1c6f33ab7fe37e230cb96063c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "7ca6215a7ddc1800ea5c3938d2449e1779f497182054b4d14163be14361c886b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "51f01056382e82e81dd1b3279e835cfab620f015b4b076c867753f0a6f693706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "96aafffe9b9dc96a64fba60bbdcb8bf3fc6b08aa6152c4a17982a07cd527bac2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "a119c36e542bc8138c8d1dd73abd38feb7451765376ebc9598962ccfd59aa538", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "724b0efd255fa42a64d8f054f301978dcb8217020e34f3da9c21a344918e2702", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "889a89dd3b1f179a20064d0502dac9f56d50d72bc48dbf6d2466978db341422b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5425511e4baa3a90db286a163a4223e3a4351a7398a2395af62e19cfa0b2e76b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "3aacf5e6404d4d1995f07393b48d0eff478cffcf278456c9615d93f5838954fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "32305b773379f754ac8a05058ee6e21fbadc456af7183b696ab926f9ce91f928", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "990892e0168d4048b8d5c713200318e55532b2212c1940f191f612ec9e70322c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9826241f2cbd7a76bd219d41926479435f6370d0e883def5ea8e182d9340c4d2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "54864295fd36c767a247250e7b29eeffff42c09088b8c3f3bf6ab87988c4d6ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "6210fe07c7d21c4054d46e315159b1e308c622aa1255ea6c4ae9cf98d1fe5805", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "18f13a412f38b207f2c489cafd5bc927cebe735737efea754fe2dcd9a6c3e194", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "aaf512e4fcf4287b06ba70ef19b418b6357aabd362530c34b1a32ab12c799b92", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "411635e55572cea83248be73956f7a79b0a464a71c9f6d10139e9fa252f07442", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b8270bbab8e6ec4f6331c3a2b35d03963a5a1fadc3df53393dd20b9ddc37e7be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "0f405826ff842e63989d9d8267a0243b0b24cf8efb704c71320517800f75f5d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "9fe6a2c762999c37e96c64b2f04d192ddc2947f764d27bfbad80a6e2e43fa463", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "9c18932c711a0acbcdd567aa7cbea80dcb2e03fa832b5000024ae9a9bf42db3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "48bebc24c388e0caf4e9b472dd39264dfa1abc7587af54aaa4a8f7811eaa5d8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "11431641838f98302f265d5eca2231a547f2ad732e46632acd90b366f8294fac", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5a8ae4cc71aba237ac4f14f4551c5a4a88e8ea717a733aa9cb4a082a31ac313f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "2d8d65ea1de1926a3f7a573918cb325e216d4a25623b3f9eaf1fce8ea25995b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "084cce82368ec5fb4012a5d82e4fefa28e37edefad06bc5ee60cb57a8ee57505", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7701aa648ea5c904610ddb3cf7c32738c69d764cec5e5040958c1ee1202ba043", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "224e95145aa0b8a22e3866b88a211caaa94969a44c63db33439bcc3a003b6377", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "507e821c982b86d77ef4f2cb9702ea8484d26075dc235fed6b25df34c31983bb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "8a3b77bd6af3f80b700b7d7904111b307ee58c7faebe34f8b0d5fc5d5c4c1b78", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "18a942ccbe21b4241d4e828330cf424f5798d45552cc3507971e3a9d10050593", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a59f9218a9a9378c44c70849e148dcaa6d10178d6883da78c913abe1c977f2a1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "8979fb606351cd91c44a384119f9b88a451d17432d1c4d8dc2b4243aae0a7a8f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "569f23f9a0f30a6e02fd290f508b9cc3a3bb1052151e347ba5848914aceb097c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "23e3a60a9bd243da4ad669d0b09cc50cc3ae64b7e9cad0c06f46e530252bdf92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "90882ea120a6a542d0b46856ed9149894fa0c067d4f0613fb3a3985946b311df", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "08b6ccf0f55967b28c1ecedb3eda5974e72e00191681e6fe2ff0efdb384c5864", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5f28d3f545b4675fb1b825a9ceb7e8aa32eb0807dbeb94249a2fd7bda80e7920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "23a49d9a52245b4a580d9d046b1479ce8427452269945a53a56ec686755fbf3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "080434be19b378223b836d33d9320dbe20bc7d1a7dc65d79360a6b936f2e845d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "82e522b3f1f0e424410e8992831fb6c253db97ed0e3fcabfac7cfa6107982382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "81c670acba0b76abd5f8cdf60b69496580935644395e649f5afbcc2341bd3634", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e2a5d8034d4dfcf85f386053c3a533b9a61c3732417e774a4872749dd9efdc27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1f0104611981f3aaf143523ba1b475e3c72e6935511b4bf95fe8d64592cdf197", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "c52b557080e7021d4e3be8bc6417732ebaff14affd21630eec6141e3587172f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "3218f80c142067fa094a9cf8150847e8d990c0248fbbe21e94e275a76bfe57ba", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e1db5dcf407455e184c29603aeab4a7a7d798d5278b55441bea3f52d930858a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "39d2407fd88ed47fe19709d49c352edcecd694239739c753bc8f1f244f3e3cb2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "9b57dd2a1e2f778377f2428e8e4c34a9989f578c857055416226d20132d7584a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "686bcd325627c28614900c76a03c5fcf01f4fc638bffdaf7a5d5c214f6edf4e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5ac38421e8322579e85f58eeec7551f81258405bd939031a6bd85750b8eb8a8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "e75181c52b8999fed56b6ea6ab711b4c7491f00fcbfd0cdca57d135e23ef2312", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "42847d912f05792e153b4c5d896bb89eb53ab09476cfe5b4e8e6e88a170051a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0b9062d49323757b6c8c464eb5c4592c7111522d4960a5562bc77e0ca82a7ae8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "8d775541d4bb9f2bdc5da71679e0bd325108714c145ddb255df19c9fe56ae625", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "84755ef0ca30aa47a0b6723c65e14f49e65e894d60f3e3f5a9a56ce4aa897b82", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "91d2feb77ab0b93a8c934eaa7787e004dcbac1c2a4239d35859db63fb9270345", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "09081314dfe1dc79ee183d209fed6592da955bd31e3eb1caa0324bfcaf3fad5c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0f5eee4e22e3bb84e868ddbb3758e1d45467f283fe573b99ed04f07026ae12a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "91c61301c027d79bfa75f8aa8691dd6ab33189fb7f070b6a0efefffb46ae4d56", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "df40d1fa94cf89ece437d5a2492bbe7950bea41aa101550976643fb43cb30d92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "23616157bd8fb6767e19de94236c8fe37506c7c3fb5b77137d51bee0a82e4491", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e0ba762c649464839f62b110dacd3f092ef5fe2b9bc787fbbd9ee4f7eb5886d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ea213dce3507ec9dd1a31cd77bec7911d57286c239fbc46abaf8288337f4e4cb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "50425cde162fcac04fc883955a1c240536344109bbf3c796f0d36710ae712268", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4188046762349e23cecee591b0d3d2e7ebc91a88b1e62a35f0e14bf0dcb24737", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "a1ecaa9b66dd25cbd2e5fefda57f8eb61d7e41a5af99526372d1ce377832f0f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1ad0c6cf4f5676c18cd741ae9e9b7baa678349a6a38301e20244ff349aec56ed", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "ece29f1493d4e4547816d3a540f20db793553c79c36d1745b4ca3025194ed89b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e963eada3d46439e2161797bd85f9bfe831f15226494fda3159f31090504ffe3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f9b2caf6315165835f027ea35cdad1ef8ad78fab897d1364e29b5ec1008a054d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c6c082b9ee19345503bff5677aff4655818b9da79b76cb6351ce7b1fb4678759", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "804357a87a4761e264c17f04af1ac593303a3c01e7dd3cfef0fa63ad61caf932", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2b4f5aa33d49fa6af98966fa53bfda807fd9abb9d6449c387e6e749d5a964f52", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "271e2b1945293c92650c4714de30d6eaf573bf438324c9f734b3f81c7872d9c8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "aa85588084a01eae70eb6e1c5625f1ebe1b0ade1e2e14ed39b969c99aafe995d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "0a50afb55de459b77d0dd3a4324e60dc17d7ac34e5ff8cdcf5ed90dc01f377e7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "946455e3adb30642e21d08fd6742694a8bc3e91749b13dcfd7d99bf320887a23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "4005fad73644d23c1d633f2f446d415f824832ef18e4c6d780ac96797d483735", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "dc77a7989f42ff48a57791d3a53d11edaa7cac6e656f7f1f2caf51f72d242340", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "7619f5f0c5486733f0f94be02a92e75b2fdbe959d9a8f06c9ad7e1dbbeadde1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "36afb0be502fe695b948dd7fa15c246b301462d6b3f99ac8bd4fa2d1ff353779", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "60b721115197a7fe7927ecba8f0dc38d2f3cf0e2656a635cac4f713bb587c4ed", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b2a3c3d8795c64378250b9fec2198de398291888f2798cb4228e97513157d14d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "4a6f88cd3dd15ae61a318769a4f3849e82bc004a63830658b2cb53559d83570a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "bb3edc03277f8ef9d70b29d3acbae68df52920ee53ed81e1bab670fce570dfcb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "2a70a7d0013f03b37067ddda098254c2c77aa47e64444186ed1f1ab1f2fffe7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "1ccf6db2b641570012f62ebb8bc13a958a0f751ab0d8e6a13a8e9e592c41db3c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a60c0f9d2442d46d90a7f47f60157ed54e6e1624e672936b87eb9aeb86c79864", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4359993aeead862ee4cf06f4107dc12779d8f3d2c875dd7f994a39e229fcdac2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "e83d18f8538ab983b330e4ac2c537d2e6c397a208bd70905140f6a78064b0dc5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "3e17dacdf59b1cfb7ae16c9d909ca8f9439810c68a2c895e6427fb20aff4f01c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0fca7cac76868a37915f79a50f5dfa153bd3d9cfd383c8ea1afcda0e5589d52a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ac8d89958be68069cb3276d1a52f927bd3571dffd8cb33809a48ee87452e6229", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4b4e18c8025f1a979124abf100df85dd962e2ce61771aeb26ca26a6cefa649c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2dc04dd6aae4fbf6ec4a897ef1d00cd1a87ae1841546455fa8e6b237075f948e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "0033ed979164fded5a129e3f08ea42aaa8349423ef4976a6eea95fa783d02a97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "394d29e7500d55fcb33be8eb4b92bc4cec55123b503fa98e79d3f8891aa47db1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "02af925bdb0d08f757cc16bd332a7691d9c3f84244468d3a3ccd062d2c7692c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "8e805712b7a15580785b269b4541403332eedbdc29fe8d02de5b85dfff2b1c88", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "1afd25b806a3607e5319f2beea9a5539eea60e5638b7ab04673063334586e98f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "eac182d18c58b2b5b1db2994af9d104078d69cf7f20084b426e5dab56d742aa4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "bf24f5ef625898e5179bb4bfb84d4674f0718f1557bf04d3eaffbc9defb988ba", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "15ac91965e84ed900dc2dce0d103cdba6cae120758fb9f8c92960a8e0aea3249", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "8ad0f316b05cd4cb5880e4d7bbf5b8b1af7c24af5fdb36d68a2a08816eec45ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "f03d82753104f45c64c2ef81ab206abd8c6520adfc76179d37e6a098f008033e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1d57bbf469ba67c11165d0ae3885db119a0528a841a2806e4ff04724b5e0859e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "42544440f98bb63f841a19894693f4f8a66b3fc4fcf7968415adec1b301c8e16", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "aba3b94d76994f2c254dd48eacaa83644ad9692a61d19f5f62e5232f228e41ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "6487a356b872443b689360adec7a98aacdc015b8090f5993980a9e9dfbd3df33", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "800fb5ebfd14e61e6c4b7f22d73f939651bd2b4473c712ef4b86725102edef71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "992f0eae3b0780da0e1e95ff61c483cb643052d306b1f79d87da14e6b0067be2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "8c8b6bac34314c6e606315b180fd745751df0d088197c1075abf59d060792a07", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "d7154013a893bcdfa2a7990e73685e9d7995662bdb30e09651e5bbc4989c6fcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "2aff972ad317fe8fc44577a8722e3f98ad84f040e5dbe90984cf449feef55465", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "5221822c00c824ae3cb3ca36352a9073f6ecbca0c7e82456d881f66ce3324db6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "58a9999557dcb1997edd6f6c6c04ab937dc461239be1730e62cc5422b4f74935", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "2a7f4cd6c122e9123e7b01576d5d444a904f47fb60f28007b4b93b0b8e12721d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7b04688ab2b14660ff5cd001dfa80598a65d760dd809a445c59f44cc0fe7ac92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "c47041e08a8908c14be6e69b90edb5e1ffa0db4d6975b92fe8a1302862fbd806", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e327c2d12542caa0ebd77a68092185dba97df8bcd8f85a90cf7b62583632ecf5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d6439ae446c9ceb9bb7560beb6eed364b28ce50e2195db20fbeccfb99301789a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ebe155e52307ec0cebd49174921791fff6a9e6c185ee57e7a6c8181ea1c79269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "bb484ce8a6c478a0640a5ade9e1dc5ccc93f6aa07728636c803ff9c0b4b1f995", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "9a39613cfbb8218e31340048e8096ab0737da5912591f2493887f5de9e872de3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "66d05f7fb42c31c7a423e6aa326e370577e059a1a63058099ae93e63dd5f9a61", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "523e219a85fc74cf4446bc19f6c47cc780d77c563a0df5d61ed829ec541fa024", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ce55d381ab486a9d85e3175bc1c85b29bbea2b8ccc5ba9d2ef306671f2e8fb40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "043b134c3f2eda2e45d9e5a46258cdbb97f7bf8102429274b30fde750f0b9490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "4ce75aa00c975c7abb3e6df134046ce98e44a3a823f7fb869c4354c771bafcfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "2745826972f6181cc404766625c78e419971275b643f7fc4805b84dc10e78c3b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "160c8d5285b50599eac47652cb7574f8b421e936166262d76b51cc47a5726a20", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "7e2ae7ed5987863930b5545671ece5155c06fc7e2fcafb922571c4d8ffd3a348", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "b699d6df61fa7b922874fb7eb802f5b0b6d438afd9216dca3b7bc4f54a984a67", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "3b3771a3667183f081ae4acff22dfe3dbff500e421fcf173ce86593291242ca6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "178af6396c92bea2f542a0fc797f58984f22bac267a2f8414798c62c780bf5a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "e1408267590fc739a48b30ef90649eb7c52ac0598cdb4cd2dd45a74c9294200a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "1ec155e8a5dfe8f9937341368738875e84fbaa6671936e3b6bf54463c46967a7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "d189479996f2c746df8c601debf8090f95a19dcb69c654dd16e86854cef06f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "4710507d85deab030e8700372340613ff4b536c8cf48a79b33677babfcb0c3da", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "e73c56b9dc971e7bc6edf7ad19a19c3349a2c36b5062ba99ac33be5f2a9c1614", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "90f8c3cc1840fbadd86b5f011ebba73098862b5db754497e89ee93e5d675e7dd", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ad3b8d85ff8922a6bddfbbba6a1259962b52fb3c7a2e6e115af80e34b055ee56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "010ea21359d9f7704eb1b8e8209a53673cf0d8c3f97753d16c41747acc4ec3fa", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a65615dd36eeb266d7adfb81374b45393e4df6807fc93411f012e74135d8d94f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5f2607d23e4162b9bc63d79445dee1d26082910a44a48f4e80009ffc22333138", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "7d7ebacf05b88d8b36bee86dd708cbf194d1a611a24fa28e6f6eff2eeab2340b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e956114a48fc4ad318bcca2858ff9a9ba0aab841011f71309934fe3c5fca0c50", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "994610ae27f3ac7c9621d5b177930b77cdf878ac3839dc270cf38125ba13fd76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "09d2bfda0fac961dd00c1e9043b75d7813d2f7f3be1e65263cc80f471532a437", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "5648bab65e9fbbf27a76b79ea78538080482bba321a36b33238132b4d188f275", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "f2824d9ab7805f8b8c4df86a0eef22c8696e8a0e86e850be07279ea3683c8187", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "badcd29ea8b4814514a986efd885d474be1f047dd7da57070e44adabae426d5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5f24eac978be1a48506c8cf15c86a52e58bd74366500cc17b10fe84c35f85d4a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "88712d90efcb61a79faea0124ea2b27d9d4718e05cefeb15eeae75b59b0dc96e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "fc25e6a0e802483567cd171a45cf5b26b9b4b13d726f8849ee46b3fed24c6cf2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b3c7562cf7596ba4bbf53fd8acec40cb21c690ff75cd641c7dfbd0f6f658bb20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4aebed2ec730d8caff36826cf299ff26aac6061e7cf9503e7f1e312bfd40ba56", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b3407f85daa05abb2aee78001192308a933f22e1320b231e4cb20bdc3c12f9f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c447009dff504f69c4a7dc9547a350e7c23c559d00264a8e51518cdebff71edd", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "83d8e8285ebc8fcd710ca989aec456c76d9dc2aa76d1dd8e3c92d613a1d156f3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "6ff813142dcb17ba85544148caf814d478df4a92b1b5712792f3208ea9f55363", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c2dd2ef3ee289f764bd3b2d22c2679aebe2f825e29d975a74f7a214bae5e0de5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "09bfb3970d7cf7481fdccf3cc1d791841b378ec71dd1ac12a287d85cecc09e19", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "62ec75756c1b9f31c455b809f9eef2f3c98afd51f481e29940d7fbbaf06a324d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "2e09c2fadd05a22af4951c6b5ac68d09261b4f237159e9458e493e47e2ed2a64", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "96e91c76bcbc62464e9337540373c28ebcc41a057babeb2618a97951c5ab28e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "98cb7843d54fd4f2d239a1dcb30995a43c52aad663157751ca109f60f1274f34", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ca740622984775159f7f1d51359ea9c7d4ccec74650a72552d01be203004c786", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "4b5ff01302426d16bfc04817825c4f8f0099334b30b1b8c445761884f224b9f4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "e9893fb35e4f8816862389814c9a8ba93bd3b86203cc0352aaf058353536bcdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "23fcea9976dfd83276d67c5b8407dfc0e4ce80bf0a79bdb08c36c7aad0564998", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "592b39f0f94f2c587a3fad1851e149d2a05747b83a50d7e81cc0b256cfc7bd9e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "98836a658bff74e19db1d391ac831df1a1709cea747af759c11662bab1ffa2ff", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "75ed59d21bc3e7d10fb124cddd92d824ce8bccf17e389152a169b29faa0addcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "39e0febc1231c3c7ac38145f7feda2a68791c209cf72b5c4d9d27f352b11e9a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "ae242ee4a298b9f05f24917f5d2d604d5c3b29c221cd30293168c9b18565d585", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "756e736300c026159b913b42355e7754b6e4cbe8ca8cba4ada27d6c634f8f4fe", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "f925db16451bfd66632893285b0192c51d69838d1aa8f13b84fdf5d5798bef10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "802ca6852ce303731513d2aeca302499e8a805b9ba0b5ffeef051ae9117a6fff", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "b69fb9f75bcb0e0985c45217e9427763e7f5e5a35b539be559822d8d2dbbb7f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a17315018e5cd92ba5ce8b59c320fc6eeb5150e5d9b18f64aff07025af025197", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "df5b1d4d2e94297c8522e686ae51d6d775c3d9c57d2b0ba5e8a2b5034d51672f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b43d1af76e7e30d63ebca80f15628cc80bc3b12e506556d95d03e5b234fa1ce8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c1eb050459a60687ecf008afb091896baa37bbf17326678052a2b70f979c87e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1b7dfaf51efd04c0e145b95de1d8717fd8b88225ae4642b82ba7b6beba90b40f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e3f71663f6130dda6cad430d606aab65b6fddee8d832061ace3f9c38f93da4b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "190e9a41525151b3e59ec43faf355ed9c13421ea5452a280b13659f639d288c8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "3ce4daa2845d1e0c5f2a9ebf6ae7548b89b6ea1d89121244f4d03c7801ae6f68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7a07115337edc0ae85bd5ed8fcda9fa9885271929cd1d6042df1378535789220", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "f5c77941af014a54e88a8d5e54b25c23890806af595ad4b6a5a6f26c2efdc3e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "a7413cbb19f59684f2231762f3924a372a8a615fdefcc5ad188e91c977b1a1f0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "298d00029b4ed0d0504a2b9fd2685c5fee344bb070d5843823cd59c80e50c6e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "46b5f682b41b706cd929ab41e4dde375670416c629d7f0fb0c62ae0a0eea6280", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f5f26d10ac776523ae525c81641e00446b7c6c58742b90ed612d2a7aa15e2d3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "35eb8ecfc4456131ae6b957855c097d8750227c18bcf926f2c5ad5d06aef2f14", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "4c7bfc906b4b9f3612ea4e179e3de1c63c706ea1b8bcaab7f5461749695b2eba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "df2f52d92ad92efc8e2c7f82eab27c72fdbf166214ce7c9b92311e379e816ac8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "43faf0432bb5603c712b693c0dd7f2fd53992eeffbc0414f42ce30b9de4dc177", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "680746d23627cfca6a599942323e41c1c05b638aefd76fbeb61c8842b39f2332", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "04c52a5ce7bd876e6e3ab2b0aea8326d3c782e07afbf3bf29d507af78346c058", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "2d86409ba60c5835097f0f65d36e9e5186c56268516b3aebf38b7da6787d5192", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "1434baf82a39a58dfd97ad7521c90b8bf2fc32d4e8a4470518a705752b615c70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "ce2e25fc25181dcaca47b441e18b5186f7836df44a94643d201c2aaa794b6f64", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "a828d27da6556ed8e3a461f34f07420b2160e9ace94e01d7f860eb3c0283c5c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "2567a0f2b571090d0d9750b199305f663717b44950a2b07a1975718cf15544c3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "0e10bfb8922b675645783942fde7eb0d57b384793a1bc8e0a61c571d154ad082", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4f4e6baf3102c5cf65f337a09e31ca91de1611cdc176ec220dfd713378f970c3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "fc8700b298cd9899b9f31c3699787aea300d2493dafe0f6c2dcd75f1b8b90c42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "7708e2a6888ed92ac35655f29487566f033bcb89ea0f12b596fdd45362ab55da", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "cf1ec50cadeba94599c8c59558285afc9128987131c31df64893de4d016a5359", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "8c7961907e60899dddfc124c51b67708750de5979ff232faf1bbe6038b4972dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "0a672bf7ec95cf56c5878c832f4da379851f698662f8df3063c52cfd24c50014", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "6f708ecb1fa54e52d6b7fb2e9b38a05716b780e4fbbe80c7456d6be47dd3b8db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "330ce8508c49f8839e671f546fa119672d30ac3f4459963713a7f0eeb447ecef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "f7e0e02394ffb69a3bad7d407712557762fe9ca5da147a4f4b2958ebd5fc0203", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "7483dd10ae33980d8f40abb3587424f0a3cf977f762ee444364d21b328d1ad4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "4f6d55303fd0443e61c7d03a0b8d5879ad2e7594010b95c72d6c8bbdb7b4f2b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "41e7ca15bcde21edb463708d27d040388f1b01655bd7c6f393c626159f285c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "0e921c3109ecbce5fea77ecdb8cd1b14deb127111d51b83a5ddf31f55614fd37", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "ac60d4cc1803fd09a08285f2b4df4b8e38b19ce3fb36eedec202a6a6ec7977b3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ab134834c837acf24fb79e8f337e8dbf6f6e8c29c5c59f5c9918b698d74afd56", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a3151a59bb11357f88fad5b9b6adcc52474eeb171e4fe94580c3923f6fd243dc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1e39758a5a47f3c601d80c4a9af9cdf9f59dc82422c6172761cec67fde072c1b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "3712d69adbf4dcb4c76ee71f843e811018d88b4272c58699ccbf8522ac0d85d6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "9b92a0790fa2a4eb9a0c75e75abb51e3286416afcb72bce9982d5c44d7279831", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "261b6f8032c31daf1663dd33ac9dbbd7b5b1bb0584ebc5d13295317af92a606b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "46abb979b7a60ddb06de88aaeabfebd0316c1e4d97d597a32c028f656dba013c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "14cddc8b1f663fd38732edd5f78a5a913d0c1d16ea61e170a485678b2b123fd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "9911a81f842f66d16e278a2beec06a1919c246b0b4ef6ad1a929728bc7eaad7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a03629ccc073a9799a4e21097841dfd1b0820d71113e802459e925a29b4377c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "319cfc002106c83699eda8efe716ade41eeca693adece9f4f528a081331f5882", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "5d1e7c0f0d58bccc45ffb5751d37ccc6901c2bbd10fc89eca2d164d9f64d6180", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "c7ac244fe5f64d165aa64dff0103de764a6b3f042bc51ee4fc56fa13ed45d3b8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, "548cc7653ff51fcb4bc4b5ef8575b2c348f7c043d5bfc483054ddeeb16676510", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "26315db4b4e88c7042f5f64741394218e6730c9454e6d23df41fff98a96e0a87", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "98b3be7e43133d82a43a09ee238b26ca6017de29bb33462b609e4cfb1d37372d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "f716bc0eaa5bb6fd521cbb46901e64573187cdedd109b805eb43008fa81f9653", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "de686ad2502ce11d20632b3140cd70897c5064b649217795514dc74424f54bb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "61d511fb3d8042a3d8562d8c4d845c3257d0a87159d62f1d4df043c2a785bd24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1a8649e909a3bda2b87ba24c3e54901c011d47fead43b08b8062577d51378c70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ef707910dcb92a8f7b5b04fc3344af75f5c124c04f9023676e189beaf794de0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f530bae78f5c97e79e9b3e2db0d17be25c3bd7979adee6c047147a8d4886d82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e08273e052f235b4dd92bbb44ed960dba5a1c8e54a23966e73db2d2319823bd0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5648de94fe4d39447c6768044b4610331f72ddfb78a49917f9653030eea9a91b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "35009c7b9a3ce08d03eb02396f083f4ed8043e9235093d620a52f62d92032ad0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b817d41380a282b7065dbef9a8a56222651dc37c204b555dda2f57413b8c3cf", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4db73c57cc6c56784b7570c7a0b6bebe4cae1bbf67fe5cea0377c38d1cf333d1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7c462e32a6ef60ef8677b8c3aae22fcdaeb16964230720411824564d52543f5c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f26c307277b0dcce51b9c8799015abfd8ed3adc033e2838ade93b805a275f024", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "de3637ba99f2d9016f26a50937c0656a14cec2bdc259413d133620fd6c30069d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "791f7c4ffc346442bae8f65472ebfe03a203151f62c560d7f492f938f89b5dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07fce295052cb63aa09dc3de5945004d952120a742dc47e817dd78de117d55a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "068c819af93d9933b26d060b2f735db530dd138e853f126bce63f07e4545aee6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53b7645b9c65b7e05e79721c5d08f797d8966e611afe2e73cc00e26afbbf1f6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3bef7dac68b04e8db2354530cc73c1b4db0a51249ac47f026b05787b7131b672", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1e5e50b5a66ec74e5b4e14b2995164489753a21046a0c78c90ce6a3c30e7134a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bf182580e8929f31ca9ccd0056cf2a9fca080d69a4005872efb51e7225cdfb9", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7293359c4ab9d44997386dcbc7641490070401b1c8ab5c394cd43822ec1a0d1b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43a98c13675cdc18b2ff0ddfa7a34e9732d0734a55aee86419295e3bb973a6a7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "13311b0fd8b5a7c7c6f9bd9aee742e0dc0d31b40f89c5d985ca1bed7c3e7a52d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d209ea0f51ac323dd3f4d3ae0db189a63b27e0f9f1b1c3959028ce5200ad0c87", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c771d706185bcda9311ed54e038dcc2b8e84b80d1df04d418be13d6fbd471612", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "cc0000252d3ce7f5305f9baf8613938f7e2ca392615717d2227426eef0839829", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "0ad821625507a4c7da3912433e01cb3eec301b96de57d77d96f758b0e81797c9", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "36a766af6b0c6f468656cf133923549c5b08b3d09a55bc337b84c1c801308897", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "58bfbd6e5589b2333d5bdbeba1519292aa269d70755097ce1ace474f1ebce0e8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c13e4da0c6014f833e6b594322a48f409b46fdf7cbcabe08c7e126e71c1aa276", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "efd1f9add01b8158c36dbc18f6aea708dbc5147c1b8977f57bae016d15b7b8ee", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "ca4a6f5fd5fcdfd7328aa0d4ddb8b4ca6ce82d18f6a00079748dec68aa368884", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "d9089bde4f7f24920c69b877f51834d7e29e37c43e438d8909c9aa77327d3b8f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "cc7b9a12142da5e27a15a8373284d3a3cf5fcb7493b372d87861099631de3450", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e89e86c567c8f69b531b56decb5a995371ba572370a72ade4bd4d5ad76f27007", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "a523fd5718cfb150681d54a09f0c749abcc752aa8d183469db673b8305dd47c3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "f219f263eaae5085b81a02232fcdd7fb13a826771dc98ccb558a49eec0b0216e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "07e4f84e440f895cdf76a4dd75b715292e16852b7203b8d53a698517d57df19e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "16779f4d18fb50144d3fbedd5714d23ec5118315105c0d17f283912bddffbc39", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "76ec62279ac918b12590413922614428b5beb285649d3cf4c0ae62e7a4b43c44", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "909cd5e2d79cb51b0259761f547143553a20d67fc17d64b66ccbe8cade606b45", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "2dc6aafd159691cda959e3c192ffeaadc07888e19a2b98fc0c6ebe376f6c6fc4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9209e6278e0bbe9e369a3b9f15d7d9748c5e969de65cae8ab2ec5e04dff2ba86", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "2bcb71a6dd7e0269bea5b3a84eb1e6879eee35fb0de0dc8ba2b003769c565b53", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "8335546e312d62953b4feb8164aa2dda0d185a9cbb5788006e9f96ef1ab6e342", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "3e848a5b7d712cb2bf5b1c51cbd94855233cafc44fb71cd85017f00edf472c88", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "4e3796885012978b12b82fd1b08fe2c89be8142b52b95caaf8de62105600df53", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "8407b64ae20d119683217ec0591f289d1e3506b9778cf6de74bfed4741c51176", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "9b5d8fbf86c12613a9b7134b7530670e4be17fa6536a7d0d682397ccce873adc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "916da66b409c5fc3cd06ed928cf569530f55e0d642ac4586344d5fa86c5030ae", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "07f2c9f78e398dbfcdd1130d19eb8a7402c73f64ec367f9e136f895d9263a9e6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a87c4c29e2b7e54522cfc2e479924592a1275f4170806377324a8abbd0e39406", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "28071478af6b75a2af3c9b85e66b2f5c6af7a81f6969280556a7dbc33b068ddd", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "cf0dcaf7206afafd6d39cda26e66ad2ca07dd38cd7d194942fc50a353cddf179", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "801a7c962cb6812609e8fde5c861febaa93b516e693926899bd523756ce51e73", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "161b315606be2c5ef97c9fdf2728e77af234ce96eeea634df41cbfab40ff0864", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "39e784514786eab432629b764216cbd0a08c9c4b74a81d800591748c4ccb2d9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "caa258e9b9e176b88db3b3a07223ee7d3101a942921118e485c02ee592eb2b66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f6a55d6615cbe6c51e29610c8ec370b7a21be7e763f069b0c46ba2700dbd63a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "490495862674a6a1c4c02434654da66004271a1855d8d3a7498c8bd63db1cc1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1529dd3a722a39b72810998fc1385992c1f49542e760f559cf58f0ee6f071680", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "90d13b6e3cc9d57b069dbd81453b2b4d512425ff2486412f6b6d38f385e5277c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87ebd806bd8adeed1c2f7c75e365da512fd4f1b892103b696e0f478747cb2ad8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7f7f912f30349dbb989dcdc972b7bc9738a70d7a8ee815a1360f8eb4bac139cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "14ade2132b77e9e7c0cb3d0224fe2ec0ae021842ee6e975a6613028fbd845d31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1bae8291849712413af83245318e984dc9aeb5bb80b38a671409d0e050623c1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e823190b262c722ec50ed9f8e1f13dff1094c3d7988ca9dfe0d92353b88b71b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "137c1e15a7ca00fe0af988f4c9b863dcaa752d149fb8fe50db42c21f7ef8b486", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a3ad64c8e5eced0a70809451ba4869a7384049ee34b92db1ba7013214bd7a393", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "128f1df2a845b23c0876e4fc7ec11fde337048e1f777b01e1055e8a949cbac68", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0989a832792e8e5147ef2adafdcd4068a2c5025474de8dc7168e9665b0deed4a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64dc97660bc0989bfa279acbcc692b713ebab67f5c6eace3b05e39cbec083148", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d32e584849f2783a1ce8d1a8b639be7b362e8565d414345afcc5818645097521", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e2726c1a167c4ab05cde9247c3cf66e2904f7b7d478b41b0f0e7647d69042d8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ec5117d2ddd287d97c5f91563168a695ead3e780172e69d60cf90a39d2c0d71b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a84ef23df8d06ae99d6c2d5a3f9567850b00ad7c3f121b7afc1064a79b26e269", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eda320f7b883a5633425f5d45282d33e5c2029e3719ed54f070db55e8687f0ba", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "47ba70df90faa2bc1945a3e2d15855e9256b378423574df51de53658ffa4e130", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9cef813f71ecfd7dc0ffbb9116394eed8e522b34e3f1dabc75bb5862376618b8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "26fbee0bed1d93096a260238e9337c878a7993f62b349d1d7f94fd0c3f07d77d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8ee22366e8dc685ee91ef9368059541da4474aab766015c3d4968de0c0b2c6e5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ab194884cacc68b7f4c42d5a62716b5cad7fbe2c324d008ced114cfdc2b5b579", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "094dcb2acbe7661bfe944e57c4e3a0f0819393f359fe37a08a3321a148290de8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "022735138960b40b6b204898df46a6cac238af8b7800eebbdcb31781447917d4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "879b59fa37e7eddfb15c2499bd60c104a292bf9152d8ede5b8e57e939d329211", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "f8215991e1f8f0a7f09b444e5df7b3cde7089f39edad835f28597452f31962bc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5af436222d0b90deca31d75832915d81e4fbe17ee4ccb8c0c720676dfd7cb36b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "48a0392f9f06dc6d31f7fdd7893c923cae6e0526e0788407d3680fe0fcc9e9ab", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "7aed4673fe0f1cae187959f85763a8aaa9eb4a3f361735a274b883882619e0be", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "3b081cf8d96b4ed3a925d35cf9056ccbe63b8058b9f791bc9bca927dd80a5fac", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f09c1fa684d92772893c9b70700f6ee8b123855e34079c7d4227c5de45caf53b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "bfef631674790c5622b2a8dbe99e3f7700174f26b7ff31450c93bef3f0496fa4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5d21293868489551479709188bdffb1db4657ee8f17add76f4c9ecd35645c710", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "8bd80fc33e3c1a290621ddb53a285b1d3517d3d24c72406463251bc102df3cb4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f6a66d48c3b6692ed99b2975b62ba252fd16ffc12e0b7f182040426e64829102", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "353ceb9749845ccb52fc14f1cf1be38554065c9d854f64392524a3c32725b39b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "ff4052642544e6943d703284b1b455dc0fffac860bd5b47ffc0b58dc4a9439d2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "af4d093a9cdc440924eb3357fd7ddf4ba87930dc03448728dfa0e8c320c04015", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "51ac831f51b6e62bee68ff2a4e0d7711b7738ad941da750cdf30356d06a12210", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "c481ed67bd5de081776e82993cf0e0f14e07cf95259a50df2e3058afc8741d25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "21d630e4e0cdf1c7eff8dab4e43f3454e3d99d5bb19331d9ede46114828c7315", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "303787855fd642fe51cbbcbaf14ca8c4442dcc4c49d4f8931254cd1f1a067394", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9f61c75c0c43dadcba670ab2741c34f14b282450132bda71692838658b6693cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44c0dfab1b5c2f8dc8cb86cf78d13f14cf21a815af39972fffa916d685fc4215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a5d40b5ec5476246dc2dd45eb488f0f8ff161e2faadb8977f979620d0ac307c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "328ef01c08f4010bbb84768d4ec23a562aed32543a0916dc4610d2dce331857f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9a1610d5bd6316a870d3f27cf6171f79010accb186557354cdb3fed6807aa025", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27186ec266acefc58a05d7c83f38f4559890a31e29d1addd7521ff8f9d62a475", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f2229a47bb720e23509d869fcf67da10c3ff366c4a8a2dee0a91544bda6f8b7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "da77f46285c5ddf25305927d4a3fc4eb35cc0f315c1b98463afc3ff470a9d01e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5a79b6660da20a4cff391542af1f2c67222300acbcc7e24a25575e3a7add9302", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f11dc543575cb8b9706a970a1268016d500af15b0c64a343736457601b8ced46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8810f599495d1a529c3ce13bc2b10bc9d19a69a05bef766766bb30e02733f87d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d35dfa57d395e0a17a11561708f76a44b8e2fd0e7199b59659fa5f01b622619", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a205c8ca3172b6e00fe8af93faf85cc12c047690a3a0a2a9cdc789957f41ad7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d7fe40a8d02653bd478910188b98590eedc2e5bfe75a24bf8ea1e92a295b61e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "23642453bcf21ece55957576a4df45429b97c9b2e0e6555f4aa9d2a71e4bb863", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "1a7b5ed29a98752dfdf36f1de7bf2fa6a8cb46466c97ebd369e7f35bbe1e7ab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a27e3ad87eacc1de9c32c14b49ebb500113bdff7fd74c6ed508ef620ff14fbe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0f6d23af387bfc2423d1fb29c247224fc508ce5767a0029794adaaa78b7a6830", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40a2c680c65f97061b243731426d047d068cdc06e2eccba2b0b558b27efe72ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f4e2913920a030393c6324c0020c9cf9e8b507438c392525d47a8b98e9182c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4c9c40ec22c6c91344a26e76746bf81272cc486e5f9675585bb0a7d1c4cea5e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6efbeb78bdd51b7cedb414881adb061c756dfad48eed924396b4aef052629de2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "301b86290a0e26811f7f68aae8d0cb8d1783bda3ecad741d58c75fac14b3c8dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b782d1339bf53e205aaeb0984f7caeffc14a05b61e46b13dd83ee0d7c327f82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c9a73a823a392d426eb04037a6e4f12cd60d66b1819dd2b375763cabee3cb9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b95855600f0ba88832c5826021ce6e34e2e2aea98743fa8ee626dcf75922fa7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3993ca55567fd3c7896551ec70daf7f5db31c79eedec9e7026110345636cfaa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a48e505ed9bc58f444f5d161b41d19014e5a17d52eb35dc1dc1be9e5e7e6cb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "08e50b11547d2c86814f01ef54128b3a596b45d00a8a46b6a769e5de0f553048", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "010e45fb8feec999774778f2a80428666b6efb013603e68da496908aa1541628", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a33e1b89cd4b643c103baefba07003a9a0e54d82d2d825a23c5e36802baa2a57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6e0a729256269c35988cddb7b18e8a9be19ae515932bb4020105287cc42f7776", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "30cf80070acfed85fad55307ecc44be9a404e7abc23d8afcec8a24da814d818e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b62e68e5af4d4779bbca93b0ed8d989caadc72d3a9eb76d1bf9d93e8d1cffd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "aa26a5c8714896d0b2fb55855eccb0b93a9b61570603274c3cdf4de5bea2c992", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "34ca0dcf045199985e97ed130915ae4fa4202903af4ff5d6873136abb84966dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a1f6b2d85feb0af2fed6fdd05cc23d3f0ccfced2a5b8dce8bbd3c146387664ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "66f21858b08ba69b4139c631a43f8305e65fd2053892e2e82eb27986934026d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7886738224a42c63ae14087973ba2901371f6c6d5813afc844a946e0d8bf968c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43cccaf22db1df02bf2efda72f0017c25dd3e761f69e7cd13fe7482a586cd00d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "efd218b26463e034d6f649e07ad8a22b25dcc543230f259a538b2fe7f4145699", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "619ecd8f3f404983ecbf2d0f6be16f2726016e3f8e9e2f27047c4c2b56865c68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7ba06e56b51ec614209adc3a6344d5834283b7fd0a41dd74e33872419558ffe3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "a5693425478db25466720b018de729d9850bbdef9f2515735e25a32f05abb9e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6c2110a7d516947f4c6349dd22002b2b870dc7822679102e8a895d748d633613", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3fc755cf19cd6ed1f907480adc0e955c56bb23e09a78ad54ece0eaed1294c0dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d9690f7ab835afb5402c9a460233833a475fcb1fa48958ad6776d8c6ada42435", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1048b63ee02a48fd3aceb4c1f3cdf950f3e69a439fbaf818828b641ef4a15fc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "77f5f904bde79b7c411dbb865fe77b35f6b4f53bc9b6151786d98ab9d0b36585", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f4855cd32b1647db3329e2be8680b4342f13cf360028b741587831647f64c460", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dc6de3250a4e7a8c56892ecec9bcaa1d44f2310d12a9949c28df0ac613f871ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "32698795b388eaa0bf128954d453c66a7da67e384c1f9fad66eae12a74ac1103", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "6f4311768efc0cc529b3a11d4cc03fc7f5eb16d1359580137de38b72165357e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e8c9611c246cd848208da3a36aab6f849c2e461fa5dd7102e43e0bd336044d87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "5fdba6c025e5c25ae33068b0d6139dd6f5351e56eafcf1a4ce63af6357904bc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "0e2f4c22a422b86a430110a0aea881182b1bf38cba2f324fdf3c394bc2124e3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "b57c06ee36b17df0b1bccdada4eb0d6990564f6899d4dcf30acf0e269554bdd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a096e9cde17812b8c7baf7fcf63d5d4fbd9bebcc5ff7f7fdc644beb89f748b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c95d061a7c27daeecbee07a65a7f97120d8c525ec6de7792f650d06d76ab3b22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b0e843a514245d041d833109f6e2eb129a2737ff07b0c0ec83ef655a768b9b6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5c0cf62d9d89904ab715b6388d020822554d22b86b24a12b6cd48ad55940b8c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "86f46ae1296eb8cf96043520940f50eb7e92bada11ff35a24985e13b58d2c4d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "15d57d4fac1a41c199e9aef381bf09bceab4eeea3ea90ee1f4c28ea1c12a9b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "c985b8c4bccbcf287a8ddd7140d8e2594195bd56656eff8b439a21c7ed9334cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "d771b76178dee659effb432a8243ca25455c6ffa143928e078be687536d4d69f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "452f026bdfa954848b6db69b63b84991c581490973639f1b8b32faaec2e71d97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "1b4d9852731a844eabc3c15a6c303eaf079427e1978864f8b3ea66bbcf09fa81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "85e48ed8d41985e8c5f80c7f7c5e58207cab7291a9629817d95d22156fbc41ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53f0401e515b432e0e93772668b4da5b055a39aa4f43747a9b7e5f2aa4e78f49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e896bb0b5bc0f0b9bb51358a2b3a639aeb8d166ecc6693b5a322374543132e9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ee1a9488c2373b3a19cc47e85c3edd855f007d2e9c523b51fc87bd0185c70dc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2ef44756b58523188c2e4a3def139c1abe0417707a007074d828db09e9a6ade5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "c1ebd986cadab7cb43c2cf5518b2b069bd92177a9676ca20ff19eff37450947c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aee7a7c580a288985e638c50b510454e759be90c4315bf92f72d18012bf1c87e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d8cacae92625290793a4b8e8badb8446d62f8e39583b7d09b9af7e26b0374928", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a5d2ba1d6944ff442742863ffcf124194d3e44e5948a207f24080e98e56238c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "6e971d91ef4bbc0fc27e5579086dbb64d23c61ba3932b94a7b5cec9fa85ea152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "72c3bd08ef456c400869966b6034cff2b3ad85485d9a34beb86a27c91ec77508", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "684a7c87430283a5459f68ee44476c45473d14d4efedf18145296d13c8f76981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4630b36d34573ece520f42ae29be100ba2b878721b1fdd4e0c4ce4f7f38cd0b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "21a5626e49a9a5fb9bc66310517941b2c56dd4781591987fc9bc2f238e88fc16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f6d8eeba005e9cb21f93b899eee48ce48e5ebbf710e725b8d55a2c7ab2ddd319", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64d6a9f99268e466cc02f95c5d0f180ae0bbd879feab6fbb2ca0c74a14cceb09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "211b9fa03cce02246ff1c5686fdfc0c352cab5f72aca3c3ffd59e101b93a5979", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "637c10bc574bdf5f06252f4aec90b0f8f312e771574145e4043943c41768b484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1c281571bf912b8b17dbf432c6fff414bcc0bfac20338b394d39d00877c6aa23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "29e9e8ad887a3a7f1494fef2044e579d01a99fa25e988809a3ba29e8a1e6c8ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7b694409fba3c35d87ead1a417e3b2b2f4ce1eb0dd0678c7f5138ad7fb67f726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f582b2c58a08ee1fe51500351489759b6191327196ac8d75151d48da7ce381d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cce78fa4b7c90398f515cba3889149993909a3c77e6088430b8cb1e0dfcf3fa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b0c2d0f33021f54acc09462589a836cd8fa84e1fc44b4473ed212cdc33288129", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36a2d8f7f3d77030383450a235a10ef2646fa7f69edde96112fd07fcb0a5f690", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "876af52fec08eb8b4cfa7a62607611bbdb4a306ee078cbd04e3dc881503c3d3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "532cad1b3a0b49715aafc4c3e04ff1c64f8183b34ca26c3dcbb98396d803e902", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6da099171d1a01b6f399bfc8178f8cf8803e72c7e46a1c3102b4c88cf029a8ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b81b3a5e0a796709083eb21e9b0b2177e1219e51881f8554326699e25d692764", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a8e34add50421074161a3d80ac886ecd6a1a7da2901e191189e65fe5e6b43b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c8c34c0683c0f2cc6dc96bf7b8df60ce3cb38f6fe091382ceb4cd8cfcfcabc73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0d4c94d3f62a8213e52fac0cf34c51412e23740159e1c1b337b079b26a2d53d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ecf829b7a0be24ab9e33312e0908a29a1a3cfa67f2e491fba7026122c49668e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5332de02bad25a00068c33f1930b58ec47c399467024625e38a47ad0a8ca7a4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "acc86d3bcbbe679a72b3aa850db5e42ed31f07170d7f76cbb46f2aee16c55402", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ae62877f03d66a78a5a0ffddc7cc484224196efc5cb9dd148fd0fb0b3a4aff84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a02870a880cd23893e92dbd12be5f9f70e491048bcd45d883ff0794e1a6c2c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b4af4383763c610a35656087bd293db79bc99f9f3550d8b1b9cc029060d5cd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c662ae3258d364014e1f282bd4638bf841b9a640391ae58bc56ebae3d8259c2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b785cb0187dba6a8e0093959dad3d05b358f9f39c0ea8e6b55c012f790327ac7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fef4466fabf36eda0331cb8bb0cb1f12cfa2e7616ebf6092f30cb6e4d2c0e709", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d07fc81d9d73ab3fe706b2a7b006ff49261e95ec464528e49cba177676903d6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "531c16986a77674eaa5d3379a6b929f149791c2b7c9a5bf40ce724d2597d01d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "67f19c3855ece6cd27e14650f3f805c511dc3067debdf3850b81666b7c08ef05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "25e16dc0954cfb020948ce89eb4deb60d4df0289d8da030e41d9cfa85ddb30c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "52c68b54373ab7b49eab35ac2f826033d7b7a9cc7ca6d3ff76c126ba2eb83b06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "94d62e01c8cdbebcce0fe63abeaee681af8df2358e13f2bd6ecbe52e9de044ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1b475dbbef89736bad6c6604fa7d620ce6b8f860ecebfabc460670aa2cad8bd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "51384688c1a098c8dc799cef8d3d9c977502d5633545e5a1b14b9010361fd7c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "14be9fcc28202a8ed88d7a88be367662061d8b1f4f4f0be2c424879ec3d7750c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a72a777c997f071f7e4abedb5da42cd829d4cd78ca280331529ea88efe893d9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "a598115432e2da92037b371b6d36645c79e43a289e890e60d685f51350d85ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "20bce8249a7564e8fb3e222f5d29d23d7ddb44e51bbafdc0b4ed3053bf01da60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "2e8a75ed0bf29651383c87c5d02ac363f14023b890e94769d28b4a919a85df7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c3fedea207e8d7ac0dc56c9082bae1933c2dd3ccd4c2b3881bb4570453c9bd96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ee30aded70c7735a88c10052ff128ec75ec3e9d807dba4ccfdca8f7b9f3ee544", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "bf9fc4f1c67655e5f191ccb7a80def1413f2705e1da4bb127a18816d93b86489", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "adf546106635005ded4fb5352f7a14bc7f716c32a7eabb3eadf891ca5139572c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "02e1a5feba96a538d26e24ba0901f6d661a4a6ed1fe4a70b34d67a66d30c5a49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9887d70afa8f2fb1b1a337e6aea58874826888939d2c1b138af9ecd55f7bbc2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "51227777917d681a30b74bd1347f4b31edac1999a5b6316f46fd876732055523", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e1654dc8cda5df5a61c163e6b9457c6d1e899e26cab93915bc398cd2d88016ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e6349b3f6c00f5084c56f0db2999e2555cf8b91b7eb135601b83dd0754e0cdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6f9e3b8c037313f2f48d5abffe4ff99dcbc9f6afa0a83ee4d70cab68ebd67671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "be83bd3d2d67c8ce514508066aab2cc16b949e593b8a82123648703a6ef0d730", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0ac3afa921b29f37c60044b969cff5a61d5eb5118d3fd70de58cf53482d77995", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "994989e9bb80c292d03694aad52bc13792d3cc2ef7d180c2f78b9561263859f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f5c73f10b3ee7619c2cc15ce3272b82fc9b7a83f28e398b0458184be139888f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6941c402c1b234b20f91e55fee23bae1d3ae060f6cb3e5f765c439e6d5f5e4a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2b5e0b8673e9ddaf53da3f6b3d21ea9dcd47de775f6bf3ab9ed05599c94ab8c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "25f2fd06a806e170eb9547c35f8e51584aab7b4eb458af78a8e328e3c625ac66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3829f6b227105fc351166b2554ce826071b2210d82df8e1ad958a18b38c4b2a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fc944d82327ff9a7f43f13c815f4091542a76d0b17dbbe51de1b17bd67ed8efe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c084e9983579a1f70126fd0e60e7c0c17242315172b142c3ab7ae65661d1e13c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "836d41abedc72a6bcd454af488b1f72a410d83ef50a84140f9769e307e3c0c35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "77d309ac058e328181e198e6bb810c4755279e99399ed88e0e1d4d005ae7cb8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cb9409fa4f4d135fcf382f14602581fd7d130eb006edd080a9528deb8a962831", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1dac79af24f78b97c273e6f59f98b85f029264a18e476ef9fe55ff19c82f1b24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "75677fb9bb41d9f85fab444ccb4554e50f4bd8c642f1ffbc449c94919d263d53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fde82c7de8f3ee08201b12006d12c3042f264c70f69f5d06dbbe2c3632311134", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cd9a325219fe07ea52f688cb0550668b4d13a0a5ae646a8382b18bff2b91baa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "85502945b9b6f15e25de06ab74abd63a1833c44b7a1b6478c95dec219281f390", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e33879edc3e29d7f9eb52916555d475a57840093113c604eac200741acbaae6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a20dac9ab68f7ef36805f40468e6a74952a7d9bc9644c12503caa10d69c95c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8b2d5134eda15fa453f0c56664dc5c2c045533e93b6e7f093db5dbaf6f8f09ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "865716ed8ad5c04b6a4527f1a3e398f55d3dd9b0988c1233c4a6e0e1de8877a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f083dc445d32d719ff4ccd440b45fb14337a50a079971a6b7c51c8858d02587e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aeb8093af0994583261b192c8597e8d8f6b65b40425bfcf1cc07c69241e3cb39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "df2677b3cf19a70b1eb365fd3f59510d95f2b7a2da5b4144342c17da942fa351", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "3f500d0864841725c3e80127aee37f0fa0b547d675dac84e33312f256256f902", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9e22bfe27198361ff7b2e475faca4ca090f8e902d0378887cdf428268266fe55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "707f517490fce233159e4d924b4266a023de85bfcb83b7922da46a9c20a98258", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f02befc0c74872e19aabd1588d92fc5d878586357f566a146940e5f961b7a4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "320f676c05c6377d0c1135bb51cc4cfcc730f93e5b5cc12424da039ce39d716c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "0c5b55ce954e2603d2f96f2f9b94250a1af7aa4027798ea8ed4c8eeed7af27f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "b99284b9f9f6ce371a16ce9ad6fef08db452c3ff4ce9ad0075b2cebf856ea3f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "771c061e18dfca432b40f4d2e99e3ce93991d5f84cfce99b9cf1b0ada10ff12c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "c9070252418e68ad1806c6eb35da629495cb5ec047ed2a75d36ada26c79117e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "389cb01f73c24342493e46b57be5d49c86b1a7d3c45cc176629cdfd32b097686", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "67a50e28cb4250ffbf513fb34e86ec6ac4531686f647e5306a7ba84882e64c20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "c42979f2c2344098b2aefa9fe8b5297347ed05093d1423e748d4494a4710dd59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4d29d03a6bfc30830ccb63d331243ceff553316f1bfa55bf8e4dad8852811c7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c54661e370a05d3a8ddf34b0d59132d74939f09836a8f07b07d35a9d860f735e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a86ff73bfba89d7e07273360885cbda38aa745356c5b7e1ba2eecdc2f623801e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8b8802b25d99b6e4c67ea902f19e87f030a1281be0775d8fa91d89828dc2186c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cf6b47af349c7935d179f9d7117aba94d156caaa3f562b71de8be30c6f7c039d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5cba80f248a24a785257b3830b38a53937876dc3c4728f13af4a6537e7601f3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b9076cf750e4dff3ce758e93b177553b400b08407db0ea81240b5c183245863f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4e3279bee744fcd8ca5e4e3a49095b52c68a3febd83c167b8d336bb45aa20be1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6bb116360a5bd3456035f0c68fa502d7bdba9f6be75579ca6f3688e206058124", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "06b0ebceaa14d59e56096fd45e2199f0b7329143cc0ca9433b74e936b2a2ad3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e379f9cca131c92b6f48174eb67ba6b05c473c9ef3ac282809fa5dd418880a1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4dace6844da6070dbe00435ad0b2a6e66a587e9b9f4d6533ed06a77a217470b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da18520a0e5b6791dbaf121295b24d4e7c4e4a4ac579ac90c2e9169a012e1d18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f3080e2f8a84ec1bfbb8033cb5657d97d22c06da4ab9236708be68ef4f039271", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "294a1de3a99d22d9875a398df70d9a7104998ac416a67710dee459d95d2778c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6aacbc7703b9d1b05200add46e3ffd0512848cb51181a7ea809c60a688e24a98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "71b7769ec193e4075de1d8445448ee51842ff11a81db8c12b31d943c2ca4da18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f75b5249bbaf8392fbdd2a503ce200b54b3ad0af5fc5ee6473a0c2ad1ec7c2c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2258f48a5784753b75c8064c1f3b3f834f82239a24d2252eeb62c6b9ffaeba4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f2d2fa91e042a749f20c29918ca021bf36c2cb1cf8dc8b8b1d7a18d22cf107bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aa389ada58ef39b49c2895ed5e5ef21f8ac0a319514267e4e2b59279adbf9b1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6474f5326bf1a41fcd2064649b8c865bcdcfe0a9f239ad1785d88dfe8a59f04a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "19891ebfd7c950ca187b50a0fb34ff0f7396c21f3a3129edf4d42d04f8039729", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fe7788e6ef0814ad8b1ce0123c69cde087510fa5ae46625b8cafb1feb4e3fd82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b339a1024bdda52b0600ac711178298c3ec8e7066bf06dc3323ef3efcfe6b8ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "92528911aa0e3ba1779e746c0aa12711c1f3b212ec561e650d7a05cb069ed04c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6fcdba783c433dea391c80dd1920fa4e04229c7cd76bbad349bd56c731d7629a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9812e192c79c046433347d83d788eb88968fda3e8ae98ba55895897035888e8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "901f1c59f00e684f2f6bcdd27f86fbb891cd37f49e0f41ce6fde450da54008d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b951ed318089a303b151ec63fc4f2aeb8b27d3ece233bf13aac107d1d9c987e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f2aa840af0a5432e567df80132f1d5ea45439c00ca9ff119ec4abaefad5ba0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f28c60ebbd4485bb5ed59398e3e7ff6d54d8f36f239743238e3cb39a802efb90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2916af0915a4724483dfb111f87183eb3e1a5e58738477dfd36f7fca1484ba38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dcb06fa0230c44b1b66caaedd8378073b1bf7cbd31bae869e0f5a9426a2b14de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b21318802ccf2898da35916acfa6bb8e5d4a1c62941ffd4e4e0e68e4c78d78d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "81addcd4045885fbc3721f271ba7b41a6b72448bbd8b24ae534aac628acbf36b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "63ea704c0a58a74a6731507972b4de3971da52b15451c2921f086cd75b334dfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "79768a468b9ddd208dba5038e18b44a2e94d15c75c4647ce96f91fede1640381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ae03b721aab14cf866867a849d9055ef62bed264a21a63f7fc7654b7a13dd1fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eaf8acb1794786e97a407c569b1f69ed1edaaebd4d3815c82354513b0ccf56ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "59c01f8f86da32d389574c7b33774f645d6f76e8b250ff538af05260fe054b2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c1b175bb0e99416b7111560a392eb6f9d4d6b1b7af039b235fc9fd6361d82b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e7da519d761a680475c9ea2d6acdc484d373ff4314a96e664d06713b6712be27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f77ef07acc219f118cb4706bde13413e62f662f8aa1c3412dc4f26007ffaebe3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "05425811bbb812fa535c16449bfa65c5151115c36d1c9608271ac0345e7c846a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "156cef0c0f136213975cf046dde97ae2aacdd12308266ed4180cc7dcab45bc40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bb506461d6517fca519901fb0ec4d29b5efd96044d9db3f6d62a952780ad17c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44b9a1032aef83a01e0df2aacc751cb6bac56446b81c9a2fd278343484ebfd2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "02bae99ffcac2055f40611f7ef93e0674ac16b28853934f6418345c0bfdbeff8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0d4cbe650a808548dcff339f3b585221616526307637cf6580abc63874fab890", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b67ec5e95f5267bb5b1d0dff1b1f8983f8b861bafc58df17ec666228d4329469", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e527f7998be6701e829ec2c652e36a4b39fb1ca3bf5c84b72b5f6adc57de2be0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e9c303f19a9050830b7311a97927846334db94dced347590b0e1962013c4010b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a017bc3c577a70167babc31a8a2e7ac485138d872b92f68e2eb6d413404e5822", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5d7974a57f6d142b39f812dc12781f62cc3a0c6ab5207fca90571e1d6c71c136", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b621d53c81097da1e765187ee76eac1dfe5b728ca24923f1220f5e9957a4989", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e34cbdc232a97a25657cbbefec485d75669d4cc1ac4e88d00ca309401ae05e78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "334525dd01aedbdf794f719dad8039aa6673139d21bf36524fe90e7256c0e826", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "80b464de5532a1a5864dfe528afd8e891b8e89084f89c5ea280492246f398ddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "31b1bf11768753167c3ad5d80eac7f052961a1da1363c98835b197906f7d2481", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "26d03e323436fe7165e91c1e28b7a086a51478224bc2bbb259e508ffd17356b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a7bf74db58788a977721e5b8b5072264f23e2c053750284911848516abe0a78b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "952bcd7a77db08232e10ecbf4c3075db5503273f6f207de55f47309f8fd3e7b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0c24d2e757db403f055d4a780e5ef3b5d30865c22d825e4f7f14ffdf1bcd8c1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c82e0abbcd4aac7925d60114162f71153cce18dfbde630b1801279a302afa670", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10121a7c40af3f6a9ce0ebb6f638a691d7bdb67bea8a6b5b620186b7ae1237e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e396ba1178a105cc1a0fbf5359b53930147118ed94feb8af52b55b96b3823eb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93fa2bb7beae50b92790d8d57b7d50b1215d223e533074f12c5aba8b69065090", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5bd9f344514b348ba86028ed8e14fa4ac52b407840c2961d7820c2bf6e96c2d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7b74102b74ba4b9110aa1025d8fe70b937070edfc0b22846b39ecbb0e08e7961", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "db66c7b92fcca06bb16adcaaad42210333b92edf1e44eb9009c7225249caba29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a28172aa0581464f6ba98fd1f3c35d9b11ca5b9e8ff61e642d58706efe54c54c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "334cc238dca94b54701c042db8661e15b4b18beb0e0f51398117708b9d5cdc32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a0e76764e95b0ef73458aa7f65a113768ef24728cf3d9229785f4ccd91041149", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87b19981da760209b20c1ee09d4b5ab18d621093f80bfcdc63d01e2192d9df47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "724cda82f26b5049efd3ac0e578031f2d4427b74bc5029a4e1dc9c1e7e3f4d42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "57d90c9a1125e771fa885d468c93a53d6dbd2643b27bf9b35dbc6e5b4538b3a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a60207fd31ce6f4935ac8677b61dce1814af2046350bd2cba400d13a3bb2ff03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5e8a4b891d4db506c0342668d37297ea11dc9bacca7b60b3a57da585f5376169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45db3f7d022b96a87296e14a51273d0b5da402ae73051e90140c4905416fdab5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "872ec7e3bed73ed86e0b60b47b3306315f42a03ff8ae36912bb02e0869e65a74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "618e3b2bb46d500dfcad542b878c987832a7a7228aad8e259fcc4251b5f8d14b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5131b5d8fdcc7a771f3fa5156e5a3e97a896c29768b961d106039fe4a6b8b55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2e452e1c96697cea4ee775620aab839892f7f64577fdfde3c39d87f66e993ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d32e4b4a6a1622884d88eda6f1f0be9cb4937de5536223016d0ab68824299712", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d60a0c08c33df5554be15f06b2f71020e3ccf56a5d4be089637c3672ea33e62a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c40deff892c2111be47a070a5c16224c03cd5f447eeb071f91e8564bbf14a322", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a84f1a47422c7ffdfc29f634d144c59e828e2acefca61794c68115aa2f76863d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0a0b871997a34591ad424c67050e10b2c71568e5a16d65a46972f9c1e78e139f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "80abe3c0966a6539ee4d3b65c31f71d702226e71e17071ffbd831393c580b53b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "f192808eba97a10e95afe1f5cdf68f1ec584ffa84192f215b21a1cf420ca917e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "687caa8a693522953e3df196e65d4e411725a5d8b77f8f7b1477a2e4c1681f87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "310cb2d1abfd12ac3626ab9d85344cf8841f40574566e74ebe5d8bc87ba2b1f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "313d764a9fbf1ed828b34c9a83fee105068aaf64581ec7e8cdf52224d7ad4211", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1086c7db47139a35882f6f53b4861762b2c27672c51acd29f5d38dfc86a5b740", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "89d102bbe1c08b37c0cfa74802d9ab0c5ed73603ff59c43d62a66ac04c11026e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cca0116753b987417c8efb7fbf89b1c4195734b3ceff0d3edd797bd4b12e9e2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64c62f5610b9c834d3989f00c2db96cbc6e2e444a6114d59a66f5785af886b23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "65cc8c2987253fae71d1349e2ef7cb7c711ca7e394b57c842375d5738f09f9f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "52f01925e721aff499632150ca483cbce7189e4d16626b5c69a9169e084a3d52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e9ae3fd9ae401eee85312e47beaff887aff059cfdc12ba9ffcbccbdf2b980f59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dc64ba61f166f35429478378016af48a997d3e27353a5f065d9b235c92b96cb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f2c40cb3e712df169c066649950c3ed5b92638e6dc0f8d915c2697c90eee98e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f12f1e7e3f3effd5a689f2de7219bcaa58c01ee654b9747e2e71a92000850d0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a2a040a4dc594e6a1056fd35a84d650581c03d992b46f6ccd68d870e5f1923ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ec0029f1557f67207f5e9d09469660044cb8026593133aed8d382d8e59af2b29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "413adf5f627ef15257d91b42bcc733607575e2dcd8896adc27eab44b3c3cb8d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a91fbce9f96011c6ee470738dc63d1de69b25ea938680d18c09933c2f409a057", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cb78bbe56ea9a4437d91c2f46d0d2ce434ba5056e497469b7e88abdf338b1ae8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20f1813e94f9b1ab0cd8aaacfae6b79bf558e76fcf3ee230cf33c81508ea51a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fcead763701366e92f1691c70d522b1e7e10d58a83ab788733f2813a16f9f841", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b019ed552868bfb2daafa1531efe28477a23b83b45f0b9fd677ba1c7c0b8f83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad2dd92c652578585c5f63bcbde62291e20d6d7938bdc6c5d0afd6ada61da64a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "23e09b0d969b72d9a0e62927db0474e449a0374dbe28b28f10a5858116f4df40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8de1e54d3368fb1d5bbd91722e675b7c62463667d54a12f6d75a59e12c936e56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5e6671051be0efef253f5a2adc674c61e19c437e150679937d70ef4727e19f78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "54edd9a57c9ad2f5c50b1d99cc0d5c83c758008c8c24430c7c99f1828d237a33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a351262f0ad4559fe7455ac79d2394959d689fc390b8218af83f72594648340", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9ba63065b84b9918b66ef513f28595df5f759ff75904765c13233141bc8e6c6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "32f303f14789fe865fadc6b3f747815890a0cc5f4318bff3c626eb9857ac27e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "567da18d9e6222712e1d19a607ede071a3644be9252d14b359774157314f9a07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fb6a94f6aeedac477d616547956e51f71765daba911f1f46061ca65c3b7eccda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a5fb0004e352ab36bf6a8d637c714cb213a362dd7472fd7f235a90921036580", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "00e23c84dce16d4ebe3ebda8c01ac6b32088690b9debf71caf82aff632bd08f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dbda8af6faa1a213b331eb8d708fbf8282997785fb66427d1f41461074b94847", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e633587064d5527d54fddbed706a7072d31de9412677cb02f9cd11fe06678ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "16db27a5ccc70cafd78a8223ccd368249c0a2bdf596e5aa515cf840990c94de1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c384a1c95d9ad13d6643dcdc6f8a141b79c170af98e9aff1f8025f980329546", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "360ef6247f865cd5026f29f1d04699bc924218b82eba47426b5131d226afeeb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3dabc1f98f9dc67e4ee109cff06d9c12b22b758417e865e2d0548db6f6cc89ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "aaac9aea94f232c2c3fc49a9c51e18c352bed38270d0b0424b234888d8e3373f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "a88d503752d0729acdeef569a056a23ef394ad28b4d3c425eb6d35e9140ab4e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b78f1a358542a8dce09f3ec07b8e4144f8f198bfb5602c9101171c03abec5ea3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a6d333fa7ba3efe2d90a372a8d7145c15f8649062e96a4e43bc459e9f69b1387", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7aa090e7c4faff0b9529cdf7991ce987631b1d5d0ca1383b0511f305e7703efb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d3f60b9e87efa545af4a53dc5a87a94fda7999d36cb2b72903c86681bf7141e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "2ed2242e3b8b1c7320ad22b27c5c28c17be3bcef35eaad824c5709dd6ae0ada9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ac33b97f9e0c12b3da0544989cf66ed34ab080b30febf0fe94f51db39f8ebd0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "fb20ecf49d91baac73b70191dc2cd9a37b3f9205c6327b15516982a34b32ebc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c3ff3eb5ed2dd66b78fdd758f56627f8685303f4b8bcc053c10419d2cc7304d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "df23a7e21e70e4a20ed07140a043bfaf5b8eb0a6e72735811ac3b1ee44b02814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "66440b44557881ce36348ba9be4595260ff4365d71cc2c4285d58d1bf24f134c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c5e285324cb40c0c2f9362a8201739049547914247ddbe67699deb76a3ff745", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6811a6deeae2a4d5e154a45b23283e25819056f66489b209026ad0c4282cda53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d95d15703c87f0ec87309109f596f59051b90aeed8c683d53a1265310edb424", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cfd26c88326b43dbab03cc231ddb2a95b82d3181fe082237eb82c3c668529cea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f8d62986d71a9c43af1c45a692702aa28d43976e1949f29accbbeef6d6740cac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e780bb8588b17ae79ecc3b5949fff1728ca7051c60ddd790564e50af4407d1ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "50db101ddb17cc1c75c7c7510fc71b60ce637d6d1c62706944fd933a6e5b127c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da08747aed00cf74c287e84beea5bc667cbd1a15553e8122927b142472af225d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40f6478e307910f7233ac85619ad1aadb376f49204457f84e99639dc25c27dfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39ddae5126b46ae81f8c1ec4718c04a9e533600fc1ad9f5eed8df540fd071f0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "441d645fc1d9b1639c89c81f06514507a6883c6c117aae9413337b118edf5001", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f300592f0c093c67be9175dac30e4d75aa9c193ceba6933c6c334f9a07741e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f6c3164c52e8edc0d805c08f721536473d573edb5f6948b3f291f2c1678ab09d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82788aa4d66d556b1511673bf4360bf04fd1856f89474a4348546f70187c1aa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e7473747e0239145ab5649baa3d6d38f7aa9d0b1913424e3d5efb1621b574471", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e3a2be6dc2a1e6e5fdba6a7f5efd18c472b0a4d621d0ab527ae6eaddac9bda02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6427ea3840c6c1d57b9fc6e2e4dde1af0e44140f6d6878085287910cbfcc683a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad5b5b6738ebb8ad36236d6df7bbe922f61bbac037dabb3b389a6120e3df79f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "714ee0dcddb87e333fc5bf449ab0e114e7d62abab0451860d54ef7f5726fa828", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "26ed37b50193352e8217f9fc25e7c0bf2e432f2a7512cadd8dcef12799ed66cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49134a71e05ea0c02cdae1776d7d0fa041b78ee82f33f737a2c4c9b987fd1e60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "17d6160b7c1daa86ab87138f8273359c23617a03a1c9d2e2bb710d7c9aade543", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6b5601b0e47b16e8a301c965270d583a3130c0930041bdaebcca429d308bdab2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0fb30779eb95996482238e1ad04e18b82c36ae9740932fa1f8b5e36ab76f7bc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "425734db3ea804ad845b36ffa358be8093993c026df52b3bbad943e9cc3e173b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27a21bd2fb5ad20bd0fa0569a7d0094817f353ef5240183f5e1f09908661ac5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4cfa631e4bc8916a76b710e361957cd1067f3a948ffe322f4cc45e4cb3dc5846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c3d4425cd6f77827097b810f17249391522fe86be77797281f2898baba7e2e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8baf71deba711a75ddf3527b2bbf51a8a249248498d9aee700aa3f8e4c563715", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "532c5203180ec3d659551620badb52f2b26a0d6a88de90cdc0da567bd2eee2f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "52a5cbbfdeea6d958fccef73db11ee02d52d61308eceff897fc2c2bc9b3e66c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a602c124c33b5dde2d8d81ec57d38cb4bc98daac04a5deb0a729403f4821ede3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c5ac55250c56aac069f3b5e3e0a53cd0dde81fd3f0a7aadb1767210e96fd55a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5ff47c40fb7f979bb8690b9507b091179d7b00fbeb09445347b57769ad5631ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "411078c1ea999f4a6c6807e0249a2b1183be170dcac6c185efe5f898d1040daa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1bb4ccea83136084bdef6f047b5e3864d0547e428273e0675cd4e4182c4fb287", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "ba4bce563f4f7eaa6e195bd8bf4017cf5fb5e760401f83a087d53d8f973c55a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "2b8218fb4257940abda5c3c2f8f03935ece3a4314e672bf2b050b6bcc9022fbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d696ba22f08c4a025810497d8af4b34ec3cce6504459160099935148337dbf21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55013140222f8b9be8dc62d62cba6c5a8d452cc42d90d54c5f51bb4181e6cffd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "2ee6b7be5cd52afc9c5522e8fe5e08266d741d2952e419cc5f54f0330f9cf028", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b230ee2dd425f761005eb45ca94b3486e910562d1d763cd78e8b4642045f34ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9eb04062eb4c306ef2afb93a380d87b3d03070e3d4b9e7c94ebde3ac133c6357", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fb70616a80b424e78a01d98d7c83273b03eea175957bee598456344d4c120954", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6b26ea91543e9afd5ac9a2512e5b3908d6a3cd94e103ed39480eda77912623d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "52ae5dcf54ba905e44a7f1f7407e23f6798219ad8510cfb762c8e619906d50c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6ce41f699a5dacf9353f59c2552430c960cfc3b2735551610bafc864fd36bad6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d4d70efa912d4781b7ef2b2a225ca82afef25d3d2c726f8347dc05447e56db76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "90a34dd7af0c99d52207c83e43bcdc44e7d85be6409f18f45f5e2901a61fcb7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dbd1313ee7b210a247e80f5bb78b7a570c72ded894d9eb40093010c96c17c00e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e92808bdb590d8da9980fbd7c05137954867ea18f7fd6989b1cf75c3ec1405f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6bc969f79d1c0d0b9b93245f08352c13ee5b424b3c944447c4f00dc1de68393e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c6e1b1fe779af68412b0bcc756a70e5717456188abd25c9d043a8cf12036d75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0b00cf0d8b88f3ed70dbcdd5247a6cfacdb727c361cf9e790c31d91b14f97a19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1c71e384eb724e9381d980e39de0e56f9797604311452440410af90a8750edbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c2d5d00215e171cc65eb9d0784e7b93dc27bd670b69b0e21abc8a8c171ce4793", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4f2c905e2b0c84fb66ff4806bf50e1a34145e292f2b4b22b89af75e5390240ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43e156f46f5889543f9f243220113cb6490f807abc7195eb8e7f98c95bafd8a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0de0e600de2fee1511a107504b7a3a067a850737bab399ca59b574fa93a2b76f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2eada8ee07620368bf5fac2d6f6eb36fa8d25b2828ca94d73624be461a7ffe55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a671a795ccbbe36893a053dfdbb41608834aa67e3c167ed1aef7388d6cda456d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "611b3334496732c4f9b659ff788d8e23964294d1ef8be88812c638c4c126451b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49ba299885a955bfa93e8ce904d0edb010f9d37309ef56c4d3d8db2308bf36be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aa951c3a96a889d1af36960e282ba4cfbc565fbdade9920cd594b4f75c7a6e37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "34011a134ea3aedd92253787ae14aa6a52d53f0cd6009ab6ba678615e8e8b128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b27d8521da8c85e79f60e8bcc12ef06b453af045d09f59c41786d37ade01ad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eb1ba18e5afe7fa377b9c95cfca5c3e079e5588c5a9effb8b35b01e494163e46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "91d7404ce653a9878029c7c7fbe8cc6b0cb77edd1a3e55222fdfe728d3b11bd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10463efae0f497a2b10e956268bacc6fd2be65f5e54d66dc2146c627f491d2a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e4a604de5d7fede30313e1a3ac0abe6241a2fdc466f21ebf63362fbfc72a2d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e439e755a23274ec5ffe9c6ba7e3aba3203576819e775b50acce51f8e3af4cfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "966d472f81f8706a71b23eb2aef1f16ea05757c868b3fc3ea9520fa13241f31b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a2fdd5c303c0d1ad99781658d04ade653277bc7cd634f1b5bb28529411fba5e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ae6736ee3eebaca196443f31905e31900a2848a8e7a39dbb6feb1283969c3144", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ba633b75d839cb61d68730e4d458e50ba849b8b6be99a480d003cb3271732211", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "668041cda7b9814e3691e28e5823f084e8b0b96d82aef23f4f297b39265a0ea7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20305e744c51f13d5985009bda53b594679143b5955d7d8c376e4f95d60ed51d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fdcfcdf2efc183e5e4b864712bafb76786fed607c4dea460ee4da78ad070b19f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42ee914fef5737cb31cd9e8e5bfc3289273c92dbb933fab5b3676648588d5f0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d691a72dca2f4029ce912b988384f4db184d8e82d0ddd4a291b7691f0471fc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "600571b1ae6359019b916f8ac89c5b94b0cd362069f985862bd267802fc83081", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7b134a10046376182aafa3503497922e8378e05f3281897eabce2e910b3db225", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f0015c9296b562346f17ed31bb8054c41cdeffc697362186d1747f7c3d108c9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2dc6fb66b542931439454a81fc365603dc824753aa8cc244548df0916075f34c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9ac62e82042bbfa5c87917ba385c28814e78c86a770513283a09a827314eaf6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "19c21c3d438ee5e4dcf9cee418c8a20f39654516bdad3c11f5b379e3ebc36b13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2492b54f9c5ac94c45b5ceee2f02426adb8912493b806703989f9a6f93f4e47f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c695e767f8dc5809fbdf65d68609485605cdf81296ea6f69c182559b92e9c814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6b357098c7712f8c99152ff849cfd0baade184d4e53bc398552fc4d286b6334f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f91f4c13a7cf80af12e4d44dc0801a5a5b2f3eda07d1e30cfca6279bf77cf148", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9dd237d371a3ee9dd41230632128a551ccf2b1b0b07128fdf4324adfe4863b5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "194fb2495703bb45c5d1aea6a05804103a1d1a62e97a34698b3b397095de19c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60975120f58a52559bd6ac8c63eb2064308c73e51e7b44c93d4766af09e4660a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c2263315292bd54eacdaa9b4b8e9d957cdc893eac24d26979bd4b906a3d8f037", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b57de897c44c11aa690ae009d6022e60cad8e484022df777576bd3781a22fbb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5c3b35192b58636dc0c7848f4ec2bc543f5ccfbcb0d83eb0ba7911bab73935e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e7c9bfcb17b3e1f20e4f9203b3d0cc3fafa6fc5763b684477799657af29b33dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c80ffd732434fb1b95ec405b8053f627c71e86256fb0bf0de4473a8180ba53a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f7c476e2364ebad3e3cbd5c54785e65d3628631f55ff136cefb7c3a5ad1b3ba8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "663c9cc7f12b5370ec09a25a624b5c950ad6976cae2f239e7ee0fbdcd319ee2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "caac1f8576e799f8d207084af9548c3b21c2fbfaa3cd39ae2d93a1dc747dcb2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "96f89e0f8d8a8a702795533aa1d2395354556ebe5fc270e34f002ca0522bfb33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c44dc8b57501bb490e0730af3f0ebf6026b038071e97855d9e2a7a8112f38611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "73c0bead6929fad562dd944f26e61012e97675aece1520392f9c857e94eb36a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "85ecafb27fe593646a1217d8a3f555d93d072bca9624fc884320e5a3ed7d75c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b26cab2224883c6047b2845061e7dac159a511ada7a0efcf833d4e8be787958", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "89111ae503cb9f22bfbc0c8c1ffcf849b658e48e0f869c702ae986ecdcaeea5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4242a6e7dd647e3e3a149ccc985a143482b9beaed70c560341cac78cdb723b3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "887e6f0bc51083f80d52e6c3a4a18f6749d1af014b9752fee81c6bf3e2ce0372", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2174bb7f94413e075316e9422406b58bad6800dc65adaeca96e05c042c4df1ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2b5cd64b94333f8318da8cff18c24f0cb186e5b2a230a49d575e98745fa60ec4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ec11e89981862660ee5c80ac97c102c313f23957ad9f0c57e4d189000f6af8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "004e6ce2b331b37d5870112c1a3235cd8ed46e3d054979e16927b0047de3dd59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e950f0ad34e45c7081fe879e49756f6f29a2040612d3bd4af4e8d5e6857d4372", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b79bdf1971c3f83ad8f0b8d400ecfac64c48ac59c77ec00f032538299199e5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b679678ec796876e93f2fa3b77b18456a50fa4674592b233f6d37bcf7000a5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2dcda29e2e78923f1cda2e95525b8dbe0505ea0bb02b6f36dc709efbc3654eda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b91085bff7ae4a23b357b58c4835008988bb19080bde3064add987aa6ceba0e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "908dfb59a99b1eeb9d2017e643d53bc5c69ff6042ac374446b7a0f7e58973383", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "840e11f69ab5f34002bb75e307ac6f7fd6fdc952472f4cb45d9abdeed858b47c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d164911fcae11c9144b9bf9b856b9d55095ccf09f0768abc80b0339a925d92dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3dcfda52732bfc85a18ec8e994e65693c76944ed4b7642a13f98f7584f7a317b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "113c87c1dc47b0d89fb30a259aab5a8175c32114f70d4c4fd29accaf2d1670df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0e065a8358be60c05c81fbbce1074ad2f03c941390604f7dfc87be3f1830b250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "faf7afd05ac5e812af6e002d6b791512c3a9c0110991d82664a8fc20efd24fed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82ed6121ccd5b2329d4aa4493ef6575184206ce42b1f036a46bfcd471a2f168b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d9433d3a03f02c607b55cc73394b82520a7218d943543ba6e29bf0ef0c85d46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a17296a1980c34cd7234a2974005e709ee82b8741fd238bba6df90935779648", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d109a182519743aa757c546f084c886388183bfc4ea03c94b9573442ad09bcd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c351cf60b510e38997553d71a84cf1a303180fe49d856e90b89f2a64da5083e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "47324e5c87ab6c26503be89a7fffc5bb5694c2248693ccbee3a3b46212aea77d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "952a098dbd3b62b84764fa9d901de7aadc3c2b07c126417c0c785b8cde5604c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "1c471aea8e99b592214f443e74510335dc17a6fdeeef1f8f30257c7817f22d1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20dd41f8e478660589d22469149fd303ec0baa28e897b7280666146fd6d81c4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c289402b775b1cc2906f218c110a5adfec08cabc3f3093cacce9cacba4a253ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9edc4b8daf5086d407e96dd8717dc62109e422b67bd50d072d9492aea1c108a5", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "7a9d89cd8633ac3c8c501ccfc7163c47d55d9662d2e24a1623666489568dd93e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eafcd0ac55043b489ead8125c93a6482309878dd23d38451765aef1f17fa6e71", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9b03ab704c56f3e4c8a566f9cad5d2f1e6e399b5b4a516f6e01c4218a2bd8a27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b845e0cc7c8d7d171bc904bc0aca8bfc7bbbbafb89c44c09fd74491a3f492a89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c9088bc1c0f4546757d7cf9549a3304b546cadc209e974de9d65a825ed3a9940", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0dfbf47e4a0d9865014ca10f97abd4be03672e4ec54dbe830f23daaeb1a1e5b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6dd362fa07e47e6f48b04549956e8b6680dfa56691ecc9eba42913f62caa8220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9cd56b9d019850c992198167ea05e521131ce0d726153defb77062e0cccabe74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c77a6833586e32c8da3b58a2571b065ef6680a8d98fbb9e7b2c16fbd174686ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "19e9ab7a01e118ca7d92cb27f398624839a5bce55ae3a82f8142d7db44c1ed9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d4e43bfc6839fcbb93ff56a00a2d2f71aceec638054b68987d2a01ef51b2ba34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2887d0184b1f19c057e7bbb5c58159365d3217b5328a22d8f438fc6d6b1983fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad3be82c1f14055fa181fd0c7832a3939072def55b4b20fe746317fb65d469c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b205b5b4b954a7aadffdecb06e5a2d80874353dd667c2ad6beddee184afabfaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f6562dbff130871b743c110b98072d08ddbf77b9ef41a76309f31ab7dfd2d2fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3b8fcf86c0b18120a547b12037a9b70aea4008a8c2ad880e2b7fbbd6241a5e6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "75b8944bd8e853872ed2e7b4a8849256562117d484a502dd2c2f3d7f640d0a51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8ea93f0fcdec0ff5daf74d6c6958a38b70b6e8a8e776c4a3cf6abb7a20155b06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6bc7fa2a8abf5e0c8ad271258fdc467f54d42c51b1b918f491a8f0389b5fc0d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b926340a389b8e4aac63c9125a577787c489632e2e4ba34f0c8fd14dc17889db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "70d0edb7933c6e122aef0a1ce4ebf07d7e17b9df04abe1944e590ec5fd010c66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49d0b380229443a633b32f73e00173712a69c0a31de3a9d54de688301e70c4e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82b452166fd86b34582ae9b227c37773a221fdd8f1ac903de4fc6c3f5b96743d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bb7a1f8c65ef81138d0e8e11eb7f8a05064b752e8084d9634b586ab0f05f37f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7542832dc95e217bed1642fec8d581451aa817cbad9e8807db74d598e19ce4cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3044eaf046154e822c339217dd3bf28aaacdfbf67e4ffa17270941984675a990", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5c7e93dd8c037f462e90db632f49d6f4e0986f9dbdaf39aabacee1d1279a7f78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "faa0365ea1a11578654804b91f07d0f2f66bfa1cbde628a4860ee508b2682856", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fa7931a5df0b1a55c441a428559ccf9247cd8ac4bc4af3efa82a68a8e89bf23d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3f7bae93b0fb9fab7cb53f41eb69f11eec27804d9452b0513040a87c89887562", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1bcd90f5ab968c2129f48119715c9c00de84fe4446b0732784a00f9517b135df", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "27f785ee5aacdceb7b02207974b6a74053256ff3c0cabf4b23effcf92b4f0d67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3ddd04d026402d5b9b128139a17a23833b01bc2ba6491a4da222ccba6ed98164", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1f72efaba7a39bc14f27d3a6c9cf978d6fa505c491347e9d2a4c0797ba42bcd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c3ab15cef60dc5e2e98b49bfa450b34a9e88d72e1dd27f95df7bd3b42204c956", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "90b82ceb1c01224276d0e32eb55cb281d322fefa0a339b2a153ab2da19d4d59d", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "769c2282f94a0be32c352e0c8ee5ece2d2ea116de2bd311683db3b5372d7d6d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "e99f84c282ba183b4e3123e9edd8585fb63e6d8d55058e9cb57944a88a3be2ae", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d916ffcaa61830b734a5fdd1129399be3cc2753980fccf9e6bae92a1881bab68", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "e31eff86eeb627264dc453ee634c626f1fec7c6b38280c4a2150e43a7b79f755", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "cb3ea92d8880c9c56206739bde0d06de983dc8f5c023b881585b82157dbb0b32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "efe49265517cd65f16461a3af398bb6b77d9ed29c241ab30badc73343652496b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11aa97855dfc49369f7671b42e087f55ae26d1c339730ff15b4cb3b65e2f28a5", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3a352eaf334afd7f433e9925583eb50a0bdf58abf7cd1f978e4ef89e76e822d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7f49fa7a7dbc0367527624c5ea597d0ef079ea6670503225b7518c83048f720a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c7cfa27a6b1161ae5ee3ad07c28baee8afa71a45142290a10cb57c43d608fc62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1bb83ad77ffb10b84878569151da0817f544523c4ee744e5034b758516d6a4bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2868bb463d72f198f54565554825c8c7176441cd7b2672f01efb3134bab9911", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "929145111346fbc719396c6e9df426c21416d6e16e76e04da169ccb661ee4be6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76d0ece2f815630678fd14ec7c32f986f252fdd811bea594b8de35c8daa611e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "98fc04ee56ee7522ee1e47587db373a071fdf81f5bea59267fd06b2724a2dbd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e6238ddd7c2d5b59f59714c0958a0ea0f21797576a82c2443f04512a3c3c518", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e485dd35e9f83927e3ba29f0b39a628f50417cbc6bc0f09e545a28f531dd648c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a657a7d416d10dcd27208d92eb0e6ef9a4413056d0d3927e1b86721d480702ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "29104ab9b0da551be2d65a2d90f33019418108aef317803d2ac57fdca101d14e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0d8074a8a5762d2c80108807087a2c047ed5dc49a9a4f672b14b9f67dc9f4567", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a43e2395e1dabd364bde57199700122b9ce888a79b0595ba1c07438ff542d707", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "06b347c6e86bc576e92b7a7054a35633a86c548ecac80656f0d4fcc069f32dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51d3bb5b648caf5c489a198a32c41d546685568b9badb87d427504b88d80d904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bd08fdcd6b511720fe17c09de5f81dc47869f628001850e420e9527dc7e6abc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca05642c63f5a0d0fcffeaa14102a419f85dc2578e02167ab1c04065baa38e78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "098d8e084ca120bad9624c24fc9df69f24d8795436cb814ed0646abb6f0fe419", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "653130eb6d9a59a817b6b570883873dbda528572624cd67a09edfbd7f75386c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7df180e4cbf18d02c878ac9c39992ac658a4551809ae14fd2926b89a014a449f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eca18271bf0171330d2c186194f9f37eb0634b7dd5d913dfb77a551759b49aab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e9cdcca2c1aba7fd00ad4c274b5b92f46db421be351a063e9ca28051cf47e0d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4aaf658b08aa29dabaee5bdce2eee156c2035ba697bd8205a0759b1fa273e324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5ff0b35224fb743496ceb1aa605c749afc8ed4bc174cd412fda3470b2da2e8bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a35971bdb933e340d93aa1bffc58e901f9f13d18bc24eb61cd2e7f658fb90782", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "444132754a0624564c84f77df72eccf2a5d872e35d1e4a94303e16bc86dc040b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9ed54b799342609ecb3d7e6495c2d9fba4fe5b74ac26add934a936421dfc746d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53924340323f510accc84a08b31d68696be08fb1668733a84402daf807101020", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ddbc71fccc7c8da35912f2ed9f2c5706624a559178aee2a18816e1c55e7b1cd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0fd78038cc5016ecc36b59d818afd6d2b47df5eed72a353b1b9adc765b8077f0", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "e3f1dd3e27fc73d3c4561bef6991457ebb6bd80de5012e54f4826941501325ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2930d7c6868f935a2edc5253437f0642dc245f71a2a19f0dd38ae4239436963d", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "7eefcf6e658ebe051e0736f3dec227bd850ab5467c618a7e9c5116a4283dfb9e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "52d94823403dd0181b60e0ed4194a27011e884ede2eec1badc1f051aba1f142d", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "99952b01c12d74107a965697dc6bedacf92aa50773d013a8784cf4a6aa3ca7fb", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1ba7b237f3268480eef3c55d2905961277fff9c2049eb4959b1c0d4b7ded211f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11e3b9162f042ff8ab2f826d607dd9a342f2cafc45d3766bb0139f6db203c809", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5ddbe2727270a9bfea0167fb78669cc4566ff3d49e69cdb1abf86d247c64cef4", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5c4e3ba8f0631690bb1be022d97fed0dd4c15bebb579efe1357d6e64b333e4ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "111ec9df95e1b01d25f78c027510d591c80ed1575fe92111ddb2b2fcdc368175", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1be9ab069abcd6ac1b5dbd947de17a34207f4b0548134e3ab1bd756385e40647", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "74bc38d1dd200cdfa00c4dfd556caa590f5f207e008e5fcfad9abb4bff020ab3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "363e1ccf4389c4bde4c7790f4cc2d88fbe2ac61a0d03aa25f1fcb4e8f8196d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bdabf0fc735d7ae99a67606d90f7ee6df93abf137edfef27381897f47c3b51bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f678e9bf1d7067d0ccef6663090f20dfae7e3173200f9f0053a9fdc77a9cb195", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "755adbc038a2c00f7a578c439e0bdbcfee574b689057839d504057c5c97525d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9af67900bc82da26122dbcffbfe0f1397fe314912ed7141a8a6c16c6ff14a00d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fcea3d233d07e252fff4f98a78d6dc610942212d2c4c365b8049e6ec1fa346fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1cdc123a88c8152d696c05bd2e3671be18a89361a3fd583664e37e1d0d41c5d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1e56baf3860c7a8b75680bcb5cfd7852e430bac7d219e02da4deb3c9308c443c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "700551a88054c0788aaf2c6277f2b7293cd14a478d41d8a57f757b531df23ae1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ee9fa1ca9f623f95209f1ee439cf552b98d1ae63b51c22f40519ce9bc49a50e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40b20cecee129426bb275a8e10ee1b7c30945b6f4a671f9884a341050b6a13e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a0d360cc5cb17a6981560c2f1d748185f82649740badc6299217a3dbf98b6f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0cd0056a962f96c7bf36e23b34b658bf8a33dc435f21257e3be3e3d97cab3330", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4305474062fbe2dca9048ff51a12a71621e4e0e27020b57f7cde1b0f155dfc3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "75ac9c2ee41833aad56015a11b737d5463671fe813625ad3027ab4a45dfcc7a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0fa69ff2f1f596f1991c80c6b0b994c1477a99bf66f43ec8329539962672d365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b7b650126b93248b569bba162e096e69d183dfa235d38946105cdd8ff1832eaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b71b2dc060fb93496b8d825ee755920b9eb4ff5a393c4e914654a0f90184058", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dfa9a73057845c57b97b2f9f3fb81669e5567f5a839bd51f3e4d7afca5bd3331", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "38431cda954d6a8aa9be221c56c7ed9aa6e2fa3c2bf941a246e252ae4b4e4422", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11fa579676f36a292bf69ad6d23889c2b050a14bfbf75f0253a015ebcb97784f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "68263ce7ea1506bc9d7468152e0560b0a7c25591da800ddd55627b62e9b98fba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87316caf165d2ecac0a638406fa17cea945c0e92fab07ad869032a311d1c0cd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "099b891ced5f70ef1436a1365072f06bfdcdf3009e2c99127cef0c62bf3edfb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "914dd081b792b51cb02ff30834c521c9c9d8c2c3fd32d0e82381099a38075858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ce3366152f693e98ec997fe1535f337bd9d0e944705f513f27ea8cd2eaa0b575", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d64b40a8d787179c5db56f75e488c26719be10b7e1ea85865b60338dd7c92114", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e9079c8394f0778d515752cec50b69e3ad79ccd0ee5655cfe79ce7b8aa4e6e73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a50d8c257a414a01c851324072c91eedc8e7623c61bc00b981a4cefda402ba9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "253ba4abf460d462b1b3cf1fafe28d575d4d030107dd136f8d0e476093b1c564", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e0b44b006ca76fb8d442bae4c8eb4d546b0ffa4746829e63af04d704232d552f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fc103576214dd2079846853a92f4542080621f54afb8cd7000c78abfb4038e63", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a2fed66edde05d54af7892784c3a93c8ee833bace8a2fbb12969ef21d7345e4f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2c69559ea551db4fe07d1b88adc31ab75a9bc5bb38f9cce738f207f30f7ec22e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c2db55136fd052ee97ce84a3e17b7694afce39086007152f221beb77e870edf0", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e6d3675742d9c2d88cea9b070bb07bcd0e65ba491193183d6fac9eb49fee20c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d296acd5f752d435b026ea331a912716ad7788cd436c02bd48ec9f9e5647074", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2ab2388dc4e05f15a5625345395cdeec3b411b1c55dddda054a653f8e2bbca8d", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6054741d3191505906073df50051491689e2ef9d8321193131bfe2d5eba5497c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "725ae47e79c036d40e95da910d409c869eac2f10de032cee33adfbd1ef0c82c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cd81b3e64062a0f76afa241a125edf89f0240b2f428c60dc166c18e76ddd5bae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f154eee47652de79bc7c2055f99058e7d45b150deb464d5c701d209fbd067526", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6343e52e7c21576fbfc39e91bb7bab9379be6a235686305ea6999104f22ff77b", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36444262a60387bd8f41b8087583607a9e57405619a630801e556c585e4ad123", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "002152c8c772b751e63d24d785e74129447f69f0e16f23e01cc745963ecf4f62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8951d32cefb94ae8fae5cf924a63f70f4452f84ebe394e5399424d034ce79e57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "72da2014285500ca8222d55dd65986dd68500e4f5913e22e18da177da1cd53fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "35adba273538388619028d8c3869b0c1c664fbe966cfb39099a785c8ee72c6a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fcd550f178c0e9b49f379c4c6e1ab9a3d48a2dd01b6373efb5edfa5b3c23913b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dec7073b21b7229d064e12b98ec6e4f33ac0b6f3a87cb6ca560cda1e848d44c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2fcec7053b9acfb3c1fca553687506fae65b908ffac6f2636c15369eca35c265", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2c235344f8c388c7790920cb91ed434bb29e45bd61dccd95be65e6dfdcf55169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad68344ba61cf1c913290adeec821ae9507e3066440d679e60ea0651c3848e01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "471f3f6511d6fe8f4fb452387a19392379bea7cfc8f15925f01509a1fc068a20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7435fca634dda02ca449e679db0685001c843321bfdc9f2afc235634b1b900ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a781765c0275918a5bb6854a3041e814a541ef3de9d664f35673c1b50e818ee7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "24dea44e6010efcaaaa48aa92bb929ce90bcb2e8d003142a9dbf73770040f7f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93bf3c4f09e3349d266ce501339a5219bb34a8a2ac22d30e5468e692e78b65d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d69a97c2fcef75e929bf37f640aa186f8dbaccd3414e443b0a27d50d212192d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c88c8d720c92cc79b1025ae9e49c5f8e3a4b82edeeede00e9db873ee0588b799", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "05a0f3733c9aecc53bc25ca395b38d421f29af26d79a1a19a8003a30b7a26ce8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f3e56bce0293ec0dab36fccb4ecdbcb6cd371c1c389f09ed9af3c34be7091eb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8f280835d4643f0783ab6fafac55263753532a1d7369900bd37eb54b8f2a6dd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e6c7bc122c8a770a7598576308aaf4c328c4ba1dfb82917fc0c9e11b3313cfb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "212ac5908c17d03a8a878356d00d6eb7c292d77dea4c87aca8e8678854259bef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76bb10a6884b60f1fc8703a08aa88fd255757361cf48d397ba24998b09960d9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b9acb897971741172d9d50cc85df0dfeb16b79420d4892c039cb11ef98a43c1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8d29ece72af3516a64fe1101ddd6f3f08fa7b646fe12f5b5d329771fc87b039d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39b1e710b14fc2fb974edcfb64f46c1bfe09e7af7831163f864179c15e023e33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8d9ada61b195d9541cf8b25df3fc49ea366df2771f611a232a6b1e90d2cf939d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c8d7674606a683fe696112ae6e56852a855878af6d8707c16d0f3726d366a766", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b0ac832484844a45216d8d82c5a05e212db16993ebb0cc25aa33288d1249dc96", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dccce98c8f708ea9d7082eeab4b12f58679c2d9fff89b4e4b036dc093ce69255", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d34d39fb6c3ac7ff8c5d32511a6c022e5cea4293631ee03613377a011fc2b35b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b6b70883bb9f5a0f231f25ece421d06690af353d0afddb43c08bbbaf4b7e840e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d838754cd8188ae6fb7952f84b25b5cdae9f6ffcad6ebe1606c1f92504e300e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c92305f287a631bd422ece56ea197fea604ebadeff53cdfeb39b65d7f060a07", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ddaf04bc7869d3374068f62fbaace79dc3f2107e4bdb2be0e88a0bb2db421262", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7399a2ffa07d04bba9749b7e630739f9d12049527dd69c9dfb6ad5c0e6dafc94", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "96e0792d7f1db032e78c84268e3f2b44bd04f677b9b366bb4b85b29cfb413c2e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "884e5a06bb37a437d97725bec31e9b8c6b55e1f04b94d564111ae7f6de034d39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "74e026b554d35f041b0605946ad5765b22f986261de93ee32b73b0c1ea20b38d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e009a446e05069a52f1bfba4701779e509e191d6ed29fb254a97fd28707b61a", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "1fc176529f372c028554b3fdbb991a0fc3f4992a17f19314f57757c4e22b01c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82007a3cdb4feded2a683d350f2c011fbf91eb0fbf6b22c17b2d1c6fbde63205", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d90309cf623580a9a37abb0f7377298f935f578a0221ff1272ef0e7e609043f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e704e38e84429113721e4019182c3ba69c4f0e1c36ae6f55b0a9d2cd544db421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d851550e7aade880b669d47d8f059a3b90c437e419aa9e1df3d8b02be6483f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "341860c568b59c1f7779fccade4c05d68a7793fdb52a8d8b6e5d5db0585dd4dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6d797868c1395070f468daee9234c4934204bddbac177112320e16b0638ef804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ac5f75f256e5902b421f47b871b1ebc5fca3959f4f589a06961434130a508594", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "12d209ed52fdb9fc7266b98c70448c08c80fda5030906899d7d3029d32a4bb1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "654167f41b86a313df0b9e6cc4b8ffef24d2b6fba80137e290ff4c51890b0d54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "db47380388928a3d5c6fd4fb3f3b843254a44cbf5100434cdf735ff385d28f17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3085a167389f9246a868f8bd12ece8d9bc2247fefd50daec13d5bac670bb65f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07aa921fcc8b043fa1b133a9162dd831dd3a8e332d5eec7a5f78dc56cba4a99e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b38813d182af5ef3d515bc79408daece37aafe90bede3968a82d2e33573485f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d174fa5f16f0d1014a61661748070af4fe7fe8ed58da91d3fb0e1fc6d1e0c820", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8bdffabef048af06098f4a9affcbe2a8e1a39e80f5aab7bceeaf45d0927956b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7cf37b2293c3152c65e60b019c1e7f1be64ab2b1f8f805cfae28b6d147c1205a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "deaf83b91569c0ebfead4d81af2f3afe2355106dbfe1f0eb9808f43d7af6d881", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "204ee00052efd811eac4ffd4cc981864b4ab95128d7756a4535b5f968972b4b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "748e8afe91142ae236610536278711df6244b4821dc096aa92003c921b559227", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0df5295b763f0776c7b9f6ed0befb6bc5d8861a06f6507ab3bde9519ae48fbf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b55d0eb6d5ee62953294cb00cf46b430f9b725aeb087c8a06e0d168eca2741f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "70e0ce0ceef0ddad16c3d2bf0850407b617a57171662368b7715bbdd3eda2d7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "370666cbccfa59bda8f3c5d96d026c87baea1afbd4f459e730672c59695bf329", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b236ca965944a5c22df07cd32a88a40f4dd1c6abd090485b096b77b7e4a174fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d211b9409d32e58623ecc15d5cbfdc3076f9a233d9418e69ab5af5b202ee3d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9fb630e87b27996b924de670cfe47c367d94457e0be1039ce67db35b48dac576", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d71c767064a28e38159754d71ff6eda7baaeb378c313a5e42a15828f00a15fa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4dbf0167bea97cf0ddd2b1d83af81a73e89c46bcdb0e5c2a892978eb82a813bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca0d663750f00f149ebbc26e523013ac1a202d6365aa2e4e552e5e48ec116461", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "cdbf7fb57df260dc058d9c765660aa30a1e8ec2c33769c4c8de6ab441d84357c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b5287dfa9071d557e74bbdf5141df99d33edf825271f076c55e8e182546051a1", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "26a586867da14493d4fb0368d248d207aa0af29240eab5854f4acfa25ea2156e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a363819026c5e8744828e3fb32a4fcdccecd49807b299b8bd0c7e660b075cfbb", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b47231e12babb05ccaecc6b4d4541de385d115780f82835b8c8d0f594676cfe8", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d92d53d097e467fa825ef2cc29c5ff64432b1f9892c4cce5599f3d2613b1147d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5cc68277c33f5b94ac63602acd1db681643bebf17e78e43c98246449da03d504", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "063367957483211eb658d48d8e001832ecd122f62adbc4bc94d44738bd4bc23b", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "042e6674a9fa2032f56bd7ca4ae9d3546a2e673faf69b5995c9fc57305642682", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e7c1c0a030140c3aac8d1ae3c4e3cff01e62ca43118bba3054e0e4bc1449222c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5e845d7c7972e8963e0ee2e967098ff7457b7e9d61a6afa7453db8dc084ed97f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5bb5b908fc0b62b17cd97ebcc11b8141f046cf5b7ebd6402abc53535b0261763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6e2540a39907cfc505f9e57014c998a9ca6ad9c434dd6a27e0027c1b05b4f87b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "522239714fb47f734f1cff8575048b1e410f38484f7b24a63a39ee698851b5b2", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b3676e436ba5b88429a84b90ccc3d3da282485a595586ad4278a78e64dda9764", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d62a8451431251cc22eafeb44d5905ba5529ba4ff06a8f199b3b763463610781", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4dcdbce4e74d47423cfcdfd7342544468edb07f17095b8eef503cc1540ed7493", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8941f5e967194a5a8af14dc29169cbae78cc811bd446ff40e420ceb879d0e4ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a68480a866bf2681bc8e5d8be62a391221a4aab4f7df57e837f78c13f5e3921f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "021d40c183953c83fb68238a17ef5eb0d4391ef3348333e21e2ee6047233c91a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c60e1d7c57640f804840a3eb0ea01455235c628f185a76809360b981e0d2e005", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95e79ec64fa6bdfc129890c93e6c737390982f65aa4b227aa75ac8da07a4ec20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "616c6ee7ead5647fa54ed217d105cf1749e7a1bacf28b6d411397da8a4ee8d22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8b765d8c0c16e11b0446a5921e7ddf475b27db8c5f4ef5aaf364e02e7b1f030a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "054bc65da28198d16ad4b8c655cffed122f841b7d453215f448d42286c4201b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a3df8f62a0b01cd975562d07327a54ae105f394bc850a14f94b3cd80feecb43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e97a49be84cbeec0fdf5402a8adf636db90a20e086391964051e41542b49128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0c543c4633a5905c0f3cd91b020755689ac6c92c0d57c30367ceab5a47c2a348", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4df4230c838f3837ddc0e27a10ba2ce1ed8c9ef76fe1a459f4e16ba60e52fd6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f2f47c5808d0b45325846a3dceaab59dd46f6af61925367ec8adb18c617533cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f1f64bf0596a3e6cfe8274a2bb3bd6b4986d4d65c9530fd6b66c62fe552df7cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b9232cc5253d48a77fb5ce8487e9accb83744f5adb4799b046b6b598140ff72a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "04a45f3defd2406c683ea03eb6fe458f443f80cccd7c05502576cc9ad32cd342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b38e0f65ee7f60138124fed35a67881ab16021428869c2e8df08c6f5b7be60fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2f2e138e56b511dd60f9c2421d36c8c2e6cf68f3f6e44c2be58ef24d727a752", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "21c94ee693e49c91b0ffaa0bb385e9de75ea55473593fed002f24788a80e65a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ace2391a0851fe96ee17dea430f7ff71d8c5533821e33fd6b53aab2ad6b08744", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e12378b685e961f45ad4364aac31af902b5008b5fdefcd404faf6d4f11df6ca4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2e4dc9f83c2a2b4787c4f1f957d36ae33957e3fc1ce60e48c992e86bb42acb95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4c02292e5219c986e95385905f9d7aa889cddf44a5cbf63395683f0104c45b75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b5ce5f8f51bb4c7156e7136bc8812181cb7563359bafecaa6f999b97426907da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a36c7e5f3261092aa755005de7e9a45e247c0420e828a172a08dec4d0ea6e174", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "7836dd0a38424da35c58a9d53ce1baccb3e694a8e5e2f96e233a9006a40e519e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "abe673862656582da15a1b7c0af1395b2c62add09ee2c1fd0f3a7e7044d67e2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8a11d4ec13980bf910cbe72621ea49072ce1cc10d88c2985a8860fab8911be8f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "d07c9217b18a15fdbb36af9c2ac9d36fc974e1b794a030ee89632719d32f4ba3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5772bfd4bdabe5208a20dc9bab5a33e7bb14be8daacc9a92bf5e3c16018ada34", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "6390f777f0d59efac24d95064436bf453e6976b4092c898c7c442a2f2a5c7751", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1e6ab800ac222649a0b8a68118529480316f369c8207c762ab8ef56a261af6bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2455a00ecb6ab7324a41cfcd717f9457ec5352ae21c4141f04d9063ecc5feac0", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "96b6df704e8f61588f60add5fd6503505eee406334a743be535e32a6fb186384", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "86610c846d7e7cb7dcf8ad3f67d9b19fece6b8ffdb7f468279250da9f0a1f8fe", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f08bee7f57e51f65d712a2d5aa164d7767ae61e763d980361cdb781171c129ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "120e151e87f29f8610ff9d0ae5acd930fcd9c26ac1b7d816b151500352f49052", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1fc8c439351e2aeb716112fa9c479ec4f3a28929171a6d85d84ac214acdbe84f", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51f719059747c33d1a2acba32595b3ee0d382d2bfd3bade7adb3697a80c1d9ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45251a635d9fb8421ba00718b8ad92d640cb09fb5174c79e4afd5b9dfa0a2d6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95a0d83f2d0f4328c13d3a7f45ce8eff0e424b01b7822d5338c9fb32347d36da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48afd7f320c0aa191a2340b0449fb8032de03d1b22a2bba9f4c225a89573e451", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4f7cf40853871b8d7a92dacf9c74c00db8c2accc0f32bca98aeaf88d0c0d325d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fbc64673a75fc483f3fe0f814f5995a84a409b2079ee738f6fa1c6a999716528", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5483a21feb76d9edacabea267cc49e695f5446e4cc8915cd1730ccf0324ac9ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a840e19c8f91e3586cc9af7053761523b19fe874c0a603730b38a4fe3676253", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2db59364096ec647c20341f621451adb74da524f94636c0c15e0a5c1b88f14b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f1031175cbfe0ec232f32572590f0e24827b3485527ea79909f86cf580c677a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6bcacc8f4ab0c6756bc2c41a6becb4c8004dc55db32752f227767cc0831e2275", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "415e17b962ac286282ddacdf3a89de3c9bcf6941747649670554c75df20474d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "69e475f405dc0b8f2dbd86f1b3e0fe6a3a117117545f18da978f3d87380f8627", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "63938c775f87b4ed40095fddb7755c22f6ae51393ad44b88f59035c5ca647f6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aefd21e742ab6bc8af7e7a909d2187df4b1a2c86ab81c1fcd61e9268e3465fdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d998a2f5820c1bd58d986d734ba6e07856116eafeef71ae1abecd2d7b664a087", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "58ca71de473b0b3fdb73e7a3a6c96d5b0a9c5466881e63194a34d5ef502c043f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "474f3c1e5d8c9b775f600249af6733299ea7f06e0a3321002e422203f3178dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d763e067710df168e1b3b0b4c53bb222d927c88a3bfa8292a53cb18b21214d90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7804d812e00fb66fa4a610d96603db97d71573cde0e6c709351baab32373b33d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bbfb8dbca6230c6094c365fa636c6b4cfca203041a05efeffdb185708b976b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e9a9b5ec01de646cafb231b8b35fe7cc3e07ec5ef721098f78840d96dcef8859", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1d37e1de8220da225cd9a8aa67d2a94c4d8f10e33ffb2f26647a17afc63e1e50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e12a256205c02da05b8b862979e456b46164345caa997a03e8913c134f1e004", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48f6800393894593181b5113373a9a3bc7d64e9d8f3800d41ac42ac33d7f7bde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0b73fed54d22e669f843bd8fd2785322fa026f5d6adef510623950ca29e25dda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "814c6c386906f4f2bd8d3c946891b3fc566db686614a05510c71cea9c5788970", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8eaaf074fecfd0d0845810e1db654ad8d7e3d82ecde9a7f8e55b3c7368fb94b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1d47dfd63b2b048d85ab4a15974ff9b1d7c14c6f89318555540ec8e8307de8e3", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "77b474be00fd611730876687179264d7acc4b815f7ac2f1932e538a04d03d025", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99212a6a533fafc549c9e5170e841bce40d1c31e4f3078f75dfa4bf98deedcb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1196dedb3e3ae2c1e3b7a972b0881fcde08f0816cf72c9e3604629c49433081d", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3ef4a5068fbebc4a670f3b7f49b8343f863f94c0f20c7b63f3fb38a3904d580a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dba8301c86f6a332cec2149857945858fa1b8bc44b1f40be44f453bd5cf8e786", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b7ead65f3696938e5ddcdc2758720540b3f978d36743f71d66dca3a8322c3216", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5d3a3a6a653feff8cec87c612d46b5c6f9a7aaea59963892e1c682a93a9611ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2a45f8618401bce345c7bc65e651f32f069a3cb0cd30946dbabb0fee21939c70", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ba528af3b60d98b72aeb1cddcf548516562748f8b261f7462c26b8f64b4b2687", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d0fffbedb840010fb8a9b453f1ffe752c50d1fdd35c7d0f8f130265b593f54d2", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36b2946113a977f0f41874e550a69e03a8d03182f5cf3a80260575f3ff1f959c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ef9ab2ec764b962bf82c21c2346a26159f84378240f98cb7857882aed7c16c05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5876720e47cc619bf449ffa737bf94ea0a0282cf13ac333a4fa2275e3d57749b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a784f76752c916d170f91f4873664153d4f2688b760172bbb94b330e9922afd", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "13c40fa55548b158ae68742b84ba62049323a310b1ca5ec775481c5705a067b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6ac7047664ee5b2db4a9243e165185d7ae0126ec7b9ab397b336ba3caf81392f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9fb4acd3a708ad926c249dfd038c4080e208cb5a52e8f691e8ce9d2f763c0d44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6fb0eab2671a8d80fed744dce26bd07f29de77812fa952b76789a5fa30c9487a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e64edf21d7ae1d8abde602f6773d7f0c8f3a8858e5584bc4732226db4791a10b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27b67bd84549e08e532fed974e6d250979b430ac240e1db737174bcf372de8c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "23854644448611157f1e5cf5463c00564c40010b702c91bfa37853a73c1eac01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "92a9c3acf7cd0785d9584032c7ce3de7abac7a4b65581692237fd73fdb391c0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87e4e4d3e5a2c054083a487fa21f5c8024e29f986b30df9f08879a218292f9f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "882a9086452b903bc3c2dfe42bbd208ac998cb0906633d29eb28ba9ce0fad97b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "97e928a55d68aa36230cf61bbf6e9498c8206ccb1189ac187f2d05abae1f5210", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "29ec41f77f2a1f8bea72718f8f1e6624f16b2ca2ad84f71b6d0b0860c78f07fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2391834afc0fb94ae3573e3b54754675e9bf502e833b7214470e6702c8cb7d3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fa07d350337346601913a261ab3a62fbb44f322acc421c7fb883ef498c461e8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d3ae74ac1c79fe8cf1c058ed6d1eb7fc3132ce06fc1cd86f1d7ff66e6c171a7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "429c4ca2a95f7d9dbe4730371129b8f4a0a4d2f5c4e23ef241c0f2bee22d6b0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "870a4ea17394c6b0aa8be04278cc6cec4ed1455a0556e4a0ef49924e82880ac2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6f19a2d94612cd496df40af886b34c3243a9ac857125ce3425d6355aba650bbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "536b8c5ba0a149d18cb5b240469802403537e1c33f14abd065e391f5e7cd9561", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cfcc1003691002d77e95d0833c987158055e51c6ebebba9d493543c4b9b3d80f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d3b4502ff74de6d342a304a3ee4fa550695328f25976ba17acf2c50ba24b3011", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ad04b8b2f7b8d87d47802c3057bcb1c4a75dbd84a4a64a0ef077d1dd3ddced8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c01e454157f292f5cb53ab5922c5e39b76293b6944258f4e4ea40a29ed6f9e02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b54c5ebcdf84557266973b8cfbf09f109a25423da79323da3688c3daed545f50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95bbc9f2dc86f523dd0b2c078f26a2cb46a47031c4135fadf56fb6acd3761665", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3d9197b541fb40ec93f4852dcdd8748ed00e104ebbbc69d2f1b6b904e5964b3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40b9ebb39b262ad0e00767574d6dd6a1c661c0c47816f5f7d4d5b587e1674698", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "89bd94d5abac3f7a737a6943ce96304fa45c855b7794523b44c091efa1082195", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b323ea5e14fe1536477ee85f1894df30c8fba393952455328ed67329a3b1f442", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5eb2569ddee9b675722a93edef653469123f3a15aa18540a698688b48beb9067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a12bf9638eabd1183cfb2ad30a1733b160bc9732167a676e8312cfb201362d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44cd3415b6730ede4aa199a59106b42f5515bb7d246b3d48c5af12844647ddd9", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "97ba3b38d98d168c97b92aeffbb6d15e88ef57ef9ae52c8b7258d24e33fae454", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18ae24a0a062af88cc0ca6e19cb6f814274fa774e741d52df2c5500e4e5491bf", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4d4d2388dba7086dd8cfd61559b671d80711c4573e324afebbe2baf563ff346e", "", nullptr, nullptr, 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "410f3a60e97bd359e64758d58883f9192afca9f95574bba96966d2e1e660d65b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "347c50950f37fa17ae2c5e6c81824e4f7b857edc4d9977dcba615f8718ed2380", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2ea50966cf1d1c725826836bc534fba672c894feba85af0169d6c2102ca78b1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "29d79517ec1b6a39a6520d7820b1fe20b388aaa35ffab3f3bc823e52c923c244", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d2d14fe6d263e274d64dec9f927b695b438f56025737b4faaee2657b911fa931", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b8750d011ce46817e773495ed79fe8646d9c666ab76c5ede089871d87364f661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3165fe2cfd5fe08519d83df0f125c7385c93ef0c39b8f27f2ea2d3cc6ad72f01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "bc7c9b899735453233295f293564adfa27ad0edb06b367f05005570011110949", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4973659374ad4813778222ab02823cc7861ec0f8ec492c0952ac8e7fdfffb7fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5f94c62d369bef549b6b217e4cfae6f05c412cfa1841e88b6d2d5976315f4f95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "e2ecdb4ea15a0deada088949161379a8b374287bac99ed295306a51e2dab58df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "b37ebbec6a642c6bce3c06ddc57ba71f5c43a0f5fcd091efda70e1074acd9d03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_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, "d397bc4254d4e587de212dc56d2e5e0c5ad0f196be1d8f5831320c10e82d3298", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "7cfa34c40e04d85da5476c36839d10c346705e6e45f7ef147a5ab7a719c8a040", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "0b47ab25ac772b1d7948005294642c7ceb19d6e8d69c422864437013d50a94ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "bf5792b3cfb876979030464ed7aacedfdd7ba9fb2eb1f04d84f2e75370529de5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "3a0aff848a460b37639a2522803780dfed5bd3ace8520701297d5077fdae2362", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "2ae6caa084be6515df03e8f8750bc2c4350614d1f4abc7a4b4cc8765a7bfa241", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3069bd60f015b33aedb9773d0ee37c37daa7156ff4a931d4b11cbe4b59a90f61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "5fa9f8aa2aaaa6ce31ff256b44b2a40142622ecfda9c93df743c7cdc509aadf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "ab4aeef6ac13f80528d61d4eb58530813f24db28c182cc68e98abbf03e3b873f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "ce5fbb62db6b916c79caa98f54a9341a7615a645010e41893ff636c3a2861d07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "9e5bff51413f0400f0f2f5eaa16ce8af58f6ea2027509fbaddf87ecfd1053ddd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_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, "d7b9d4dc845284da70289b42cd167902771207a2b9d3d5740a39feba5c379341", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "e4ae8a44d6f6f3aff6c984f29c77453130ba6dd557434aeee7822fe4a030dd89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "ac8efb03db52bdd291bebf0edbaacc0f6aaeed98cc55f9dc36d7d8fac21d4223", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c2d0e0839c5208a17a704baf61c6ddc9f9f39190be5d8c50ee3fe064078e757c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "88b277c4bfc58815f58d02ea112d9613f30169f23648fbdffad40049e29965bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "6a52ce3b0a542cda4a563b93a1cc40857a4a4c6df5f5146b0c7b8f289060b941", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "d07dbd4f0ea1fa5533a81b0ad9705f2eb282380fe956d9b40bda9b5cefedd542", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_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, "59d42605532f93537b66d2460fc4b10a9a4a501e011bc5dbc2ef7570a0e7757f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "e0fc7eda58d7ba205849cde9acd42dd61dcc08ccda842a6d93ac6370e8c64bd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "2827b4d45a1cd19aae48af91ba2276a52b7140f9e10d7a4027a3a1ba6c9d9e21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d93712d90f5c19a49abdd4c5b79e43a00f3c1f9a37bb5a4ae500c0e557435a61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1da21d643726b64896463385dbff24c39ed50e4d26f7ea879171dd2ca5b3e44a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ab4fa6abc30e820725b28781cc018b7bd6c40dd327713b91624bf475688eb2c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "7ac5283c32003348b6e298c5ce832b20a4ec35f755c4cdf98af09cbaebf94416", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "921e6ef0faf276a2de9365f7b68433fcc09c9d3c80f1e16055c47f0f1546a9f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "784b05eae463adb4001b8c33f2fbe0e78bedf15ecaa8f8078060a72a3041b83b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "54a98003d4bbb2e8fa4dcc5e34b6ab5567d0d2c3d43d71484394edc68b603572", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "56a0c0eb7634586e76f93e53bd06224f0d3958add1a971f032d0994124bdcbeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "a5ddbb654c27c02654dea85f352ee55d1ecc023737f06515262bcb512fc3fa94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "f05270a1354f02b87bbc4989ec1d9411af3daf3a9d8a5450b9f87289c3ee702e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "e148af56b669867880fb271654f2af069b99d50db7ffaafdc830e2537f9fc0a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "faad8e06b3e4ed840b4b3cfc06f2780c0da164a00ccb82c1720a999fe2d2acd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "2a1cfb2eb247b4671402f331cc4f984bba4d4aaed13e1cc3ed8e4bb7229a05cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "9796b6daaa368d8e5fd67e1e96cf9a02d827c3a520aec9c43a96f893c27c0082", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_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, "a7264334369bf24533e4a5c7c3e6c721331e84f110151a62e69e78de5e515aee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "69cfd6b325aeef307c7553eb2ef3c62bcbdfc5870406900ccf242949d7b89325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "7a25a48bb449c88f1771bf7ee12051e8df6ffe521cd55578d604df9ab7493ea6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_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, "cdecd4a66b202aafff25f59c3cb70fe6beb3fad789623f3f9b87b5721c3f7036", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "b2c9a99336531fa1139f9069bd798197d037105c19871f6f4390b0177c4f6c5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "c0efa4d929d0b5b1f6afc14aa6d7b9ea45af68df433c2e66fc1bd9d323d09a89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "c218a02202eebc0b6dc9dbcff2ca756f3c54a03c5496d9435bc1bb74e3603e7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "121b7aab65a4023810feef20952bedfed04bdc23708beef3eed2bf07465c4776", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "d59377d90728f4e0b368ca9474801e1842a154cb740b6e223437fd134a80c3aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "9cb77a92b092258ac6ee5193d9f52fa2eecd333286b1fd735d2a92a88fec6476", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_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, "a779270bebe7ed6f5eddc5802e4510fff79285356d5ebf724e53858a68f7cba4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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_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, "3e695e6d0e190d371b6f22d74bdd91eafb6a92c21e583397ca391e6ce0769378", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "f30302460395d3930a6e79eb483b3b6a4c54235e66e3fbdab05051efa1b0c508", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "75d14673c1a0fcb85e8c7f079691750a8d3b5b87a97a680b9f5620283649269c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_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, "3db1c3d153bc2628fab30bcef442d7b32f22069161564934bd6dc7f911bc8034", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b1b27fbc1c98281139430632aa252b94b61e241c5835a2cb0bdfacdd1a6bc360", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "b2142b91dc2a0726b219f781b42d3dbe6d284a4f1161d2594e31ed40a4297f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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 */ 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, "f650cd2d4b9add877157fc35b249f56fface0104c10667a21a419179211c4810", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "34f3a9715c79a61ff312e751da39eb260b78f0272c5c827a252842097d179e65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "7eb066130a69681b65c5fbddf53824874b196bd6de678131adafed60aae8dab8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "1fba69f6353952931b6e766e613f9a8a1ebb406a014b61f2612f2b7e08b7cd11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "b6bad50421f88e758c6f3103400fc1c803e812e39b26641376af002e1e5c4939", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "0db285d33a6081392fb0bca1539b62b207b23f6de33f97df2bbe50214cbd9c56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "b9050f1777f7279cedc17f8202bcec49c595b682e286ad59bc8d6c7850730be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "bd58b5df647dbcdc5cf9b2a8320c2caea06221518bbd0997d93a52e626ccc6b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "e0960043e5667551370894f01972f186d32069f621eee3be0a620fe29b9d86a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "c520b796b2589bf15bedc185b7fdaf46b19f86eed90b79d49904a2633e430753", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "c92aa9d2a5ce5159ce3a3d5cfad7ca28d034f0cfac9cc9f01c40fb30b9c88d2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ddda41e224e7a7cdbe76775bf9ebb6143bcd329b36d123049bcf671033627771", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "34b89b771e5d12d13d28d08020a845c86cf886f389add0307948b01c146e8701", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "33c64d73bc5fcfef01be0fcb145bc86a66b0a7480b0aa9cd76598e197d7e0d64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "1d79a194fa6163787b815f5aa8e8a983b247c69b0874ff919837d4c2f604f77b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "415d062fc6a9443cbdc58cdd2a178cc53e932eb1c46220bd2d05683599cf4dae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 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, "e1126f3e525a885447d87f06b6e4151ab155dfa9a420de50b3a968cf14b0a83e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e3f3faa880856a0b471655523f8f2c98e78e2c01b899a536a9c4c598c812a8e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "6b6a493470ee912a6fae4ed90884e30f2d8b18828e161909bb1750d70f5968ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "7676b2f8aa31f66f0a0aef2b1bade5d3464e51adf94aeda433e90623744014e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "79d556c143fe6d5503e60dc2bef1c6bf32e20a985e09744959e2ca228bc45ba7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "a8d12229aa8df2b97548bb096218836e969e27752e4111237e1be0aff6084dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 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, "d83a8d8171306c9edb63667bc55e0ce6b7a384242915cf24a46283a4b300dee6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "57cabacc304ce15bb3cfcb29f7fe311ec954db0642c61b50d8bd72aa7917d5ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "b7da4fe7548c55baa58f41ff15a84f4fe8cbf200214f729ad3e03afc48623683", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d9a28480516253e9a0d8dbf748abfcdcfba19fa6c71ed3f44e5660d4ab2442ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d251d362b18582cae33c09ca4fa8c655c5a7c1d8edef4f5f99d1ab40c4444334", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f103e9bf8c26619bfe82ea5e481481647701e4df169f0b00437d3ca366492b97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "2a4579720d84773cb1519254398e725f9f05106888cdfa65e57638ec74213c70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "59861ef1df864cf1b9b10ca59788d331dccf1493b9392585efe6c9ffb843aa62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "a28bdf14815613e85147b58f2e5d86190c61f206ba42f97acb7c633b1e745aba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "5228e475da2bee4710236b0e78ae229d54f82ed10b3e831875099f22fe56df6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9ecf9ad03bdb5e8c5c0bea9b867f2142a22038ef97c5ba2737e5609e35721cc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a30f29f01d430db635d1b66d9189bba0f079365c2f16a82a5da36ee45cc4e101", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e837f6088482c3a829850f192e4c11202d03cf613cd852107393382aad94935c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8a0619400a01e5d92a81e981152a3456964d22582289d05a793350cc3a7a8630", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "48ac549b8a7992311e3dc52b0a9289fa503ea86ab43ff2fb35f9bc7b3364d405", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "08f0dcbd28f4592ebc0deb7554231f58087690a5ad412b74e18670c2eccf64ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "98a9a861ed04441f2bd34b492173a7cb306087b0c5e894b5d86445d73f861f5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5eebe5e696e5cf376d7b7e3a71fc76106fa8e16e0e64d29f32ad290620c70e8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "425195f12f2f2f846e5b20c7b1d7d00023479c34da0b18f8dac65ea440497f4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44f55eb6f13c615da634178078886f7d4e3aa17df7c63982ebd018b8a57a94a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "107aae1f9d32a79f418146ef21e4b555e74787e87ddc471efb57f2efd6ba37a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0d5e8c070cb4ab09cca82a1cde7501064f77b3cda4baa53de3982a132fba9031", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e8f1f674b4393603ccb530af8f1c4ebcd810433c0e8d228c0fe88dd18320e60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f68350a0f7a6608a63b11c7cc67a92f4029a7b97dfde02e1df44df343d1c3a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c56b4fed0506235fe55f0a70d8e0622eb6afe2ac5bcbea24da69e00613e30ea6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c0a1ef401bdf5438d6db95aa7832f515fceb12cd4e811cd8d269d0c3663990e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27a9d79be41ed7d0960ffe802bff8559eaa3bfaa431bbdeb90c0728337eff4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "498e1561f6dc27aa0a6c3a46a2d90f7d21145c9628c73003d75d4e34d3e1e164", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "446bb3f067e7b274dc207b8814d7923ca917d8d86bba2f9419c44bb36bf43a62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ce866ff883c044e6504259a70daaff0bebc72ecf8d9934c0bef6121b58a4ebde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ee44236cf2b5197b6f7a71fac5c37987ae57918a053bde8be409c85834ff72eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3e1524db89d678ac3a4b422a870af636e767aa4f596e5031e48feb6d0de53bd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a3c38a8e6a61ad7af02e8d9d270185b5e548fe4ec0028504ba386ae4075a10c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f7e64398dc3972cd75ed76a673210d0b1bf1a1e04db66aa88ebb6ac9a2f69ed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "880151adcaf6426271206b09901deea0a58d5705e82ca5961a0e2139cd0a0352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "04eae709b83a07590398ff3fab8d1341194b8178d5737dfcad4b528e3537fad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, "a5ec0cd5163008456e781a51a3f913fb4d4b32f77fe00fd3c75867b5b5880510", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5944e20b554625c20421b48b6cf99a98bdfb5f8522ec3ddde0fa2a40aeb388a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "892daad4385eeec6e3195ecdee03813035a998489b520cf50b5da10340998a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "62262761b147a085e77b3fcad0fc0c6f58e535781afbc6664b39a26775f9bd80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "b2248e53be39f8a1fa649b4bc688c16ed2de5ecdc14e5368f0c8518954f23b85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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 */ 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, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d2fddb099c2ca33ebf3ca9c4f0d08e49a5e0082bff162e1f6ec90539a095bcf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d456d440f2c68ec3e335f7e6be032909f74f4c6807362c289d20f014a91d66ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8b676a29375635bd4445f7f6b4160d8e26bb5af0d623cda32dee70945586c28e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "e24f13859b496e911299efc9ff4a95cf0e3f2270da90f17cfee42bf131ac2a19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_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, "bc1dc803ab8cfdca6ab94f31d7a0a769b3dacb265ed6040f1184c58e3bd82928", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bc374667b953e5d70565b0b1ec7ab62655f0b0b07707daaadde185620da98008", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55d16013265236f490d6100d65accf17f239825c1ea598c30898480a819bb746", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f9aba655e6928c7701b9c8c48b0bb5d8b029050b484bab4ebcda832b79392814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "fd6c53468c42c639cd432502ba16b9b28c2ce77790104496d993146b572857a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b7d964836f9eefff99f6c95ab3858d2ebc7160e11ccbd22d60c42d67b0db4188", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1d3571debf5bbe1bbb729cf04721bb02cde5bb90bee58dac5c70ecfcfc9898d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f50825720c0457f7f32d7edba1937e6ddd386d9ba23fa37435ff6e0e9d99d642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d42be775cb0efeae48d838e06d9706b554dc8041131f47b4d47ad75a88ce5121", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a8354f17b6e9c7471b24c6ffe5d8c0b5b624f9ac541697be55cb75cce1bc4a6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d950436f77f43f7514af848fabab840fcf8edeb9d26021044bb5b0501607e632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "dc0168a0b4e258a00931686be049542471973bd58b6b27ec687d6d55117ef791", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2ad9cb213eb5e5c5385a901075fde1d48932ec7d0f24fed25f877d83dd3fe6e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9e2232448901552cf5fc04fb96fc209fb7d6222fa39dfd7270be4c13577bcfdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ef3bd2bc2ed3c42a4834b50a4d6e5b1598ffe76330455ae8dad8d6a0873bbaef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a0cb8d449eab872e499096c58054bd08470d4933d2693316687a4df6b2f41f21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "82d7a34e37edd43e8d0b3c2adb9822bc4cda3d818484a8728f5ae434eb5d28a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "bd9a586e92a09ee313a58d0794c7229075deab348928cf0aacb12cc2f4372163", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7066574880c27d466e76abec5bdff5896e2da2264903bc45d50786d5c3a6a4ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "fc7c75138e991c5e9d1c95c3d13a30284493f94606b25f8f646e52b92f7f92e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "821778e296a98775605281b67338d3f8a710e46f1c10e7ea7f5298f1d25f3300", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "35326eb0b1b88fd214e71529ff1d1349958df80643ea7587e03f190667c74eec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ea42c595168bf97fdb924666dcccac01f2ada1b65996f44cd9fc0bdd6f22024a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "77cc9acb684814f6ee1f5cf775fcb46038477b03d6919f1748422eb6eff25c6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5afa0148be71638cbc05a32f0e56cbbc3f4b972e7cdcd411a2e84b48b05e42dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f81472a2801b4fc72cd01260693ffc424eae1f0d2a707050030d3e42ff95bcf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "779d4684dfbe3548ccb9a27f94e8366b913ce693449c17f8b0b8579d4162352d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "47596468ff1aa41c115efa10b25650ad41ce65f9950d64e63ae8eb3b3b121186", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4c0a68f17daac8539614e97477be65a49ad9e6b53fa6cc1b4d4291c071d62586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a61aa00f95f4ce1508097fdde854b9533e5551d982745d10cdee735a6048ce66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4de78918882f337bb0c2a19f3cd047ea5e60b0e0265440e12a583abafdcb7560", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "320c9ba1029cbb4d9a4c433adeb901f69962de4a75077ffcd562295b02c1678e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7c98170b953f86fb68e60382b43fe4758ff8afb2c6b14d3e7846eaa23eafed67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4cfd99fc2ebd16c554bc27da27bb4d1440bce5e5f2a486909aca00202e62e102", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca91da82c863b0e21deee0692d3b9147801af05bb8e7513ced5aef791ca43fd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "def4cab9472b6391b7cc8cce16a70e757c26e351382d5f80f8f262e3bf5a9941", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cf6a6e08d2d63cd6bd92a915e45d85f624b958b74d051ddf1d54ca3729a16842", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "41bd3c9947a6360551171f17dfaf5260c56797db24053cc85304adb5e60c7273", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3d398aa892e9263533e160f217ce51f1e8b4f3a3488868a811ceff4fd3fa8d4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "35079ed9b5fd548288adb3a5f0008f5b040e6af509c090d18bd0264a5c648282", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e050ea03f7214e245ea9c941b521f2f6dcf2e1bc144d33e12320c3a6bca11dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48cafa2ad3c5efb523aaca31caf8c7b54dc091d292135ba13aa7617d4217d9a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e7c850f4f31df4cd0b91f0f276c369cba92ee3de4759b51ff71cd51c6c1c04c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "01c5e2543ba0c2b5744d02bb2d0dd3bac24f182aa8cc95cadab041780c9d1231", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "75db37c69d547d29ede3e5f9252a9d3410544c5facf2982c8875264717aa2b06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bca80c2502385d478f6392b64d825683ffb47348bc3c287538d9b06238551b99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e05ba4dd630c83c1343a4947608d4244a697651eb6a1ed94bc9467e8f1ec130f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93d7a7bb32817a43471d641589247ea3d080bba0928f718648245a370a79877e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0ed07c300ad26421cc7821d629fc98b4e907e01ae94513818e17ae9041587146", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "311b4901a8785c12bde408c21ec68d14f4d1b9ba23646505a3b7354e796a5e20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e161732da366132afec045a8d0d243cf23afc2644a3af780210f43fd03b4ccfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42b016c0fa5f8b8383127f99c341e2dc4578d2a0b8a9e36faf55104792c4a945", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c4733739c8104ef06e25cb7fd3ecd495d63530622ff6dcdb5fc7fae0994de513", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "09e1ca296b26551c246af0644b0eabc18f757769efbf399d8acead27056399bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0289ffdc1b564d9507db1cd3cc3b09459a0046c30021118810bae4d0816cf619", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f5cb1b56c661a0b5f4d332d24bcba0b3f210eb0ccb5bf5a569f55bc9c19fd871", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cce0d9e03b24e1e7cca084c450fa6410addf64caaa34fd8fbb1ee7321cdb0099", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "22f51117da7101a08c56c0284d7fd4e1b8977ea49b8082d7372da47c5a8dfb17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0dc5f1acb1c78223bbcf7dc2f1788dec3954cdd83b40894ce52647b212ae989d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f74be5a178ac14655a5b52201eefe2d835e359ade54dfba5fae509cdad096c8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1ac6bfa61665c4f0c4d9be6bddb7b461302f009940fcb10a40f58a904ea4ef1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3b6d2843e71870987696450b7bf1b7c4d71219498f2b9ae8cc0c494d5b382185", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7f83ee3097cae587b4572dde4132ad1986e778d99e99a25f3947cc07aec549af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f356fba0a8bcce61362f2ebbd30c853cf592a17253f3e47825722b5400a29cd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e666aedbf3011d8ee14ffcecea95b9c77fe51634228ca300a74126fbfdba3434", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "57f0fa1c0fbdda6a9aceb9edaa7dafbf1bc4e8e072a7c47625a44089128fbb15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "457d7e9258ee3bb9eae449e0079f9ed75c20e8f965badf449f4a4618256f098d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ad88123c4cce2907005511a9dbec58d27f7692dd6d8e5a268bcd24196337d27a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f15a63871a04be3bbb173e143ab5a3280866c6315ef7a65e694f5281a81765c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e8dd1f5dd71748dee525ba6ebf0dc954fe98aef9d3e9ea3d8b0c2b0d3b3c2c95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2e96836059a19694baa06fe25fcdd10ffdf0532a51c454c97c0857b8e3825109", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f653c7b1a2cfff72ceba7b89be621b862614a4cdcc6db0130a1792e4ec76c1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b162cfad23fdf210ed1bea137e2cfdd6af756adfeb36a7688612aa814a7af28a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5643b4adea6ffed93157b57886708310ef19eac29495dab2d0bcaea5f9e61c2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a6a26fde95279b5e81aed067f7ff543cb315720ed67f6f096d4f3b920c2e75a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f549bde7579367f8cde40cfe767cc420fc18fb096cddea1fd98afa612b929da1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e5daea6af5ea8a00464d8fb0e17144a1044a40f550542747d8cf932f3fef6d1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "14374f8c306b829d5745204ef65dd72a119eada50a12c313c536a7c6e48fd7af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bb5fca36d1a05ccfa550682b96c28b60195478eb6ffaaf659dbd5455b90c5430", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "350b4cf9931bb4bc2dbb3247dce1aa1737215815eeaf3ee1acbb877eb3e691e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b3e61a7f567ae50253c0f600f670c028d63b8d9762881ea1287959835e499b85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "52a8fba4b808976a95d40b975e2d780009588d14b30e398a4ab9b947027a2616", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "373833ebd13e50559c6738e4ff5af2735e824a8957291af552b73fdcf9684a06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8aef5d03de309fa86c9b8cf90e942f28621994a9cd1d1f7d73bd956db09ad347", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1d94da6cad39030c6f110fd575a2ae664babed130a581274fbe93edcf4057365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3f4b0c106fddfb689da33b149971aca82f2e913cdfeeaa692132b43b75b2f5ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5e9c4435ad1644a412ff5fa0e983e33738aa354dd731fe8abce3dfbef9f39662", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "093588d5901bac2a3b482e6981ced2ae2a8a00f44734d6118cdcd2ef860d121b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b8ac682b97125d80fade8ff8dd4385758985c6bbbab864a9c85787d3f9281ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b518f7410a75054b4b5437d2d3f5a5b52b310598e66a44b82ceffc935a7ba139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b94ab85f6caf015755532595c436315c2caf9a22cd7ddf7413b1f94ffb7a14da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8fc62464fc3cccb01e229a79f4dfd8b65737bdfdf123445cc49b5334f0e68470", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b7aa8a1ce5235d18409f13b0a655d4693f29b6cf94c1d18757a81797ff749314", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b00b9a673ab451d124316626b3e7a35306eaba32e316d5ad1d52b13faf8d1f22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0cfbad6959baed37b52d0d5f1dc766357eb1582a6727b12967a317fe8cc57b57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "22a893a665151a6e6c38c0fc37d9204c2d7a9f8f6697ac423aeebc1707ed6739", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "34a51ae62da9f1ea801e7d7f38cac001f58731b66fbe92e69196f93338cc34f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f15fe040fa928e1aa7de06737dfa94e6b6e84a24f0cba2775ed7a23b88d61724", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f0e9dd1086f0e56f17ab93ce261e38aef21c2f9e82dce185d2cf7d2a83b12023", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a02cb479acd45b6dae0d9c2bda3828bf9fe29e0e954d238e40e97ee48172983", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4056081a75232c2722225f5e73c46b5c9360d3cb57e77386edb5a6f68a87599d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f024bf92f1c8d9fa922a08137ed1ccf769f82f2530591a428ddca3bab737e6e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10bd7150b230f8f4d16e17b438dfea66a74ef5e9a4ba99460d9dddc62d3790c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "269ab83732baf9df3f2141b27556d5a9a2ecaef29ec312fae87fd64886e06c63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d2244835b0c4b01f3cc59472acb4a22dc09ba86fe3f53c876f36b960f88146c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "44be6368f1bdeec4f2bed11c0e33ed63a9805d63deda7b1c58cf02027a074922", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e288646dcd44c984bdf54f79209ad7b76486bb5559ec163bc835a62449462cdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6c54f289bf81b1622f2aa7dd21542b540d935955132105ee541a63f9fe5083d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ecf907b0954973ee146c6f6547df327752b1735cd48b17d411b65962b6789cf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "168abcfc0b1597b6069ffd62136e9d9479725a2bc7d71ed60db48b3de0476036", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "bc71c6b460daff5f084eca98a50461f98458ed70c20f22ec6baa0ba577dc2c77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "99697ec44be2e07efee4217d70634902e4a3085b4cedf66efb3970427b2569f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b8316fdd3d9bb77d631c075e77bfa26e888b4fb2d26dbf8445b56e3b8d2f845f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "8182bd6310d5f68b5257b16b6b38211c968fc966c7e0244804ee2fc6a12d1285", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "97b57ed860dac1ffa2a4e1d4a6064a90c8796e708e9052aec91de3269507d91e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "376007d627899e475d0734a78989a18fccda2adbccf178643db65e9c2705098d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a28c4c5dcaee0f3d4de0ea931e09b79fbd21600dc237d4ab5d0854e4b1868683", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c407bf73475c3067977c0e98907cbdfe4075b5dc829a0811c025399595bc0734", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca387b805cd80db365c8ad0b786e4e59d97a3de46b4fb989cdf8bf5ba45f7bf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45c31be7cc463307f48091d1cfa5e0a6d40dc329a19f2a86abd735b7a6482e3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bb325b28c62b14ee063e5a6f5f7e945b915787b2bb292eb9fc3538290cd56436", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "af73fd1c55e7da23d438d1a46572d3e4f635ad5cb5d272e513ffea00b7bb9846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f487964a038d09bdf0c97be31035e55c04bfec4c4ee85e1528ad519d32ce4dcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "995ed4f6a0bc2ace06fdf0006101264534ef6b6c01112cb33135838a28cc76a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "56ad815f8b49cfbabc51d6d787a5305e77f226bc610f867be031849ee0c89f23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "61bb2129ca036199d8a6a800f0d1d65fd996023fb5c82e04e2fa1a6bcc4228a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b237d63a55fa085429520833586bc17e94e30d072ab328961aedff255de08e09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bef522c24030192e9ef8b9a0e9221a7eab7643dd468a1914fda4433f08e7410e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "652c634f78173c154191e03f5545af7b8f3dabe2ac3f98229e7ea24328b81cab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c4606611d9cd39c207f5ff8a09de775175cbba29833fd467feb1172492da1e93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b87a5418cf76efea5a075db9dccfa52e6b9f44e5c9a657a3d3817e51d95ebb9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "810899a1ef95bbfecfd49d1d9ac9f527c0c5c30664a0d76c455c6474e478a87d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82fc11f57f1758ef582d70ae9350fd0126e6c5d8286602a7a40aa91fc3717253", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cbb0c692bbddb8f104bb69036b1d6dfca7b1398d6534189df3850f8b0a82724a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "213eee4fd3940559a389e2721e0c1bba39ddbde61c8f744d0df9f199adec89ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8f464f299ce2998b02c595bd54c56ff7fb4bbb1fff4311c9afccb55a2bd42bb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d283a5267f3d131aac2a7c16e7561d0cb495544a102c09bfa45b4d98f5d2383", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "633194aa233a3244efc97aa8caba3924b167063f78bd181e815a481713874f05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8bf9fe77f9b1b5a1c3b8b24ba56da867e2e99a88395fea4fb72b15ab9bcee2b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f49c5d6137a4c3f616b8d8f26c3165ee0a31d4fdc15cd5d3b31fe6ffdb3ac0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44b674988ccea17839d5045ac5332010a29ae4a74e460156d769780be2234a74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2beb67be75d180fbe24fe813de1e161f422fa5d825f2e002cf4bc22216188462", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aad9a602e39b27607a3dcb3c5ca1ca85f1980d627ebc5670b131937f0806bb42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "34886d7107628b970203d94ccea9af151bf4644b1abe638912d96db0b09d1b15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49199c2b0501e4af6f496daa6b60345a4de5e96765b0e0e2cfb2d88f0a0b3ebe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "db25b9d47953c99267f235ce2ac6b1fcdc91f65420018c7b6d748acce32a4f7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b7f2db98da9e647defd74213828bfac0fbb281209e172f6a97d7b5d99169dc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7092476b6251b6299db65fb608628c9edbff1c034aadb6c891e36013523e0c74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fb2e8bbd5a4b492f803383d64f7f19c5366e5162dae172a8cceb54530b0a657e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "251d9da788ce767a0b8928414a69885ce9eb569402960b247b693a1995bfacd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "3d0831ac8fbb196b4de5e4e78c30d51dc9e36551620529004bf9aba2f0c73e7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9884289ee3c99af348fcc8242008301a3bc2f572d6a23404d13e378eccf92abc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "830c9d60fca62245c002bee3431c564ad9537da7a2f660c4620a6c34c43f9086", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2b2595dff6f53ee65b8b64ceb910f137fdcd4d8f772a8be035924e7bcd797dc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2f89799d1e95683d5dc9a1fc10d3ac58badc16b898d916eddd4c6b1a104f1b58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "caa2f2d3f5706a843be50e48ae6627fc43b0e1d16dff3394a2b609b1b0ee28a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d083d602a5f2288662eea49e9b7fda4792e5ae6e25f2a0bec960efeba11116a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "24ef6d0964a7793794d3290f8b8fdba90b923f3c564d9615c11a706a72fa254f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "84870c2657027489fd02718c9dd7e568b2585dd2b742155f319a7d99d86fd539", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9a4ef6973e980c25d9552017125f2997078feb76bf801ef542e9b67161adc0b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "012bdc6a571296dbfd3052da041762a82dec5d6fc94dfc09440e416ed4cd95e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "3cb627ca7282a7dbe1bb24996768a607391e05fa62b1cf84778722116e63794e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9bdbbc298d0a8e02a09a1f41e260e31c46cf53ff64bc923687af885a992d55b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7d31716e442a2f3a589b6af93c788df1f6322e4f70b399ad1f74bd5f55ced40f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e741d0f3e668b32b75cec022375398e556c62af9fb57922117ecea85421151f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a999c162383b34c5effafcbe7f94b61212967cc969087b40db5b27086c5b9773", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7b5c255ba00c04c4eefcfcfbbb835a0e3c6b5ca99d001e7acd9603c5bf63b4e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "33e2f3b00c8cedcd89223708c43046edf29e45f3af1726b3da94e915e9ea230b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a9ae0c3038d0ad30cb1c2d53beb17da154a8ab4dcfa3c58b604c80e62d369b3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "f410adde44b610b7d3bbbf3752237d3e747469833b85eba9038f570d4abc1c5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6fba186c2694644bd9ab393662c6facda13ae843b6a44b76029b40f2b752d86b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "c185edf08291010c9d4168992ac106aa9daf604a130157817a3b0549a2668574", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "ce6fe2d3b2c9e18929a045f85e829bacebbc8785ac3f72a3214804c83e4fdd16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "0f2b2d487e79a038d296fc30d109840b7946438d42c3d9e7914c5c7c923e95e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "4baf257b83b9402c165a88bf216dc17b9b53f076c093425477e879eb2ed9e40a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "8324c95991c8ec54a2d1bd9f86e0cb108be4186a8f2488f2c428a7dfe5e51fbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8a97518e1d6f4af000071def223b2847e640ff3d80e70d8aa41064a6ab3f5287", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eeeea2a2bd579fe6b9709e5420a97e55ffe007db174ffb22d1182d337105ed9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bfa933de159d61c653d2420f97cd5e4eb78e7a9852df4bbeacc52eaceaf9e572", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f57795dcfe3689f3da0070c140c6347aeddf1681b7571f2f5a6c131275b5dc0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a37127dc5dea851c44505f2196b6f7f53935ef43ec6f0f5874af1c22159269fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e7f3ad12d6db03ed8b88f7eea52e93efad89715706f05387bf3544c731aa8418", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4e290f7417ba106a2736cda1ae377035a1671639dd4ffdbe221ed80980305280", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36f0710bd9b9716b62feb7b612bb7d068cb81df4b0c651299364454d5e2b37c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, "2b25cd747c4b2947bc750063d7f1e63dd2fadcced31366861d4a9d96eba72896", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "41f880968e19508785d4d931c66376eede7580912191d53a69a5457ae4ec61bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "90b3ccc3a90cd75f13ef649b9ea6b7ee780a99d5432e52f6f981f1b0ca40b409", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3cce4589d8fae4fc72d4cff127eafaa105bcc0f2daafcab9d55722b985b5a77c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "569e8abe6e3157404f4b88e75afa8776ce5f0eb1db7c2648c4e1047320364882", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ca0f61c217266cbf12a0b3572689f36da472a409c3f102f517f890bd675086c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "4f1b2be3fd1c2fff68de62e229bc6b7e53d5bd67f4bb3e35ed7eab4b619494b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "06032128976530df95bf619feb275bffc359f4b5d68095fb3fb1acb886aa6728", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e6a0a8db9b7a0e110042d6112f630fc1ac554e15253f57b70982b5fab056e8a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "81f6ce9060173a87ff17cc938934b2c92fd657bbece6e7bcf79ee82f59ca22fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "18b5683030b95bf876a831cf465b28a6df6bc0da4a50b046e896e5c17286da0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "092d70b2d207472cca85eb1442bd770eb8246a2a01eb7bbae501694e047a7e39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, "7d8048486335ddcf4f815aed1b350c8615301b99af0fb6c1e3bd12a6d50aeb9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ccee65cc7a47befb86919a5be03a198a906f99f09948d2e447bc12598ab506a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "f21c7438cf9044a0133969286c596bf3cc7903574ea9ef7001ece8779d5ff15d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d52ecbce0388cfd029cf1e1704d1bdb0ba5bc113e5b2cb63e67759c513311446", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5c70e12502cc273f8092414a0e034a7b3cec047dd1023c42c8e671c19a49674", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "30f2be4293870b46a66880bfadac16be04923bda6320ea39d21ba1b8849d5448", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5555adc2cf4caf11cb030d5316402a114befa4e438763a239e7c4076e92bbfaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "f2242ae16f3e4bdf2e8f8165984cbc0b88285a1fa1e725d4611980ebfc3e545f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 512, "edbf8066cfd5ed32bb33b2a291d92bb048584f5684e4d8d8ed46332beb241bc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "450de0b5bfd715fae6eded95837cf08be736b18e20004eee845d69386a12be4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "e82a4a9dbd1da76ffab74ed4f5a07b31660ea897f7ab9997d0453e54a125c76f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 512, "b348446b577417652827c2176fde327dd160671019132f63755b4f22dde460ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_sm103a", 512, "2ce52423430ef88e2d34ee1f49707c7a87c6931ecba4833d61b9a221d94b1990", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 512, "a8e6864cb85fba3fd6dc108c2a1c060bb260cea373f8c63d7223b0dcb19c8c62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_sm103a", 512, "7a69441b8cc242f87f69ed487dc4e75cd3581bcb7b3df94510740e3d7451e131", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "41d42c57fe794bb92362ec3598d70a32321d0bc0ab82a9a7f8fe93abe9281ce3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "29e602b588598606090992285709e939a889d10c26bd34445cf12ef8600d4491", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5095275862ad04c0e8c1d6285c305c3f0e5f213edecf0091932bdddf38b9d337", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "846e0208c1bee305fe8a4745b9421b8b9e61cbd8fa9452b410c64fbd9e2e0448", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "63643968f770ea53cae37409881bf8271b7f345b19854033c255b0d3bf53426d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "befd13f29590e05ffddfcedd84933d056795e8b4304059512261340e61064cc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "ad7505117e4d141b0f2feda4cf746d7a56c867e0decf317272fbd11ad64b9ceb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "e265c7d51995309b899a3b6c7decba6af055162cd20789260e8bc38aea008dac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e87d633b8efdc72fbc6bb8b8d3fb5348e84ef0033b81a4d7c35bf4125f20e8ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "bfbc7d9f3a014ee94f23569a7eec733b42e9f32eec1e459a4b875fb6ffe090e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "268bee213dcfe3c4a88223cc9572eed7fa801a57fd64c77d9e00f5605b3ce0b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "64048a0dde5035e1f624c732fa1e91e5a6c8993bbedf3a0314bacf1c6f2442b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "724fad69226c3cc0fd3897b92e5c223a226f207c8e3654ffdaa57e81462dfc96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "979f80f11fe6888839afeece84c9e511ac0287cb63b37b01f67fdd0ba9d3497b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "afd430dd6608ac87efe31882cb529fa31409e4b8dd02b7d33508d04edc59e415", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "d423cdbbf3b369cc0407153321c5add2cf2e3dc6069ce7725d6d4ac9c957f3f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3adbbf1a4ecd12ffe49bd279dd57a1ad7c1a8fb398c0d063a00f7eae5b443e4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "3a26cbfe5cdb5dc43ef899da9cba4bdc046a983045f763e05dea3d8c1ebd8a77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "78b20862bbf566770d64aa270e1febbd8faa85425de5583311daccab91c86e00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "6dd920b85aed7f39a505b6ee59f21298897e9371b6290118eb64cc80dde74cd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "892ccd5710ccbd6346402a436a7631545dfe5840efd9e04f7691e2e65d1fc3e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "c04cea118085ec95118fb8050bd06610d0dc0d2f46778055d464cbd6443a5976", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "3b74eaa6438477da6b8c2681ae43682b1c15a49c48625cb8b17d5a7ca0d9f2ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "69518f15a055ddaf9d54135bbdcd8b43efc04e3cbd2ebf87fe7820f3ff379c0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 384, "650592ed7a07349742ebfd5118c8fe951fcf133a9b5facf3ed79239d01b063a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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_sm103a", 512, "45f521d29153f2404f199255411405b7078880d8fa69df6b2b205a4aa11bed84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "d53c9a4de4088171c396385474508fdb8a0ff4ff677d9693ceaf8b6385d52f81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "d044d282e5a2a0a4e00458b20ba12fa1c6ab0dffc42d977d45bb2da72e1336b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ec9dd0a1ffc1c9f886bce8dc94e379bd931187075df64acc9d7bbdc6391c01cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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_sm103a", 512, "bbda6216a1f6e4adc5b552e5ebd3fd0b939f94eb02e5a23fd20bbcf69dcb9e59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_rgTma_clmp_dynB_sm100f", 512, "5ad01bcaf1b67b5869b69448f5878126569b5bd95cd3621522da30da34f4e96a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27762e2602bfff009eadb1f54fd9e2eb421ecee3d3a29a9213b45d288a34d341", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f506a477a2bffadd3b2cf8d7fefc241e5e93be50ddefae3857008d1eb835c72b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "a729034a53da073b9b7c691fe6819388230c4c5722751778e0e4e68bc3f4a92b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "601a634f88b9ea4604acd340f3a64970abb5fbc4f080780ca4d210bd06360ec9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0c181e6c0642121450c0dabcdd131d18b468b00d50d9a732f32030550a9fa668", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "33369ddce1bb40229789930b1b2d13279e996c34f4b7db6f16a216a59f672957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bb3bed0753e7ffe2da61abf45cf55e311e6d87ff3bc5d03464ac65a5b6f9d0cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "76066122974ae5af0b40fe81efc7f96c3155997d5ce7fd3bb40e3cd691780cce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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, "abb6b6b0378df4b152d7953191ec66a38b2c14925c73c6dad5036b20c49bb9fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "7aac778b6fc14c6139dc337a0818affd59d3cb4f119f66c882e0cf30c74b5467", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cbb2d41646f3c6ff0f62f048b50d4ec71d4b0662be2d1eb0d9510c629261420f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b5bb67cafe4b9b81e601af91f58e5d900d9d9bbca351e3f26d45df7c59800b80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7cc826af7412f9458144c023b7cf6e9de5733c1c3ac3e5908f08aa3e0d2facf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, "bb99b6a4b729f9850b7bc2902f7494a699e98fef711fdbe21491818a515918bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7c7e7c5dbaf3236954f70303d3748817309edc0dd7235f5114470ec1715a58ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "03208e324707efaeab49b97377ce51db2c9404871acee6312063489ed9ab8870", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ee2e60ce2627708eb89eda268d456e2662baaa73bc92854c0b0ec6f5d0ec5804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f61778ded2a7fd66030a36957c8e0b8c50d44af105f41e7f49122e3e360b5aa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8d0c5095c9ad91c76f9e3726321c958f267cdce69dc86e8eb5fef564a247b2ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "f25210feac032cb09b7c0e0c7a13a0fdbec7c84c2e010cf3c2774652f72ebf81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "5e73b0889948d8341a867fa8aab35f7049000d34a4a7fff9137fee2ba600a23f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "437567f9d2051eb09377de82a327b82c3e44b4f86b413b13f3b538abddac0e18", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "5ff54eedd19bc7f3057bea1b3ef0c410c9fe4e3a622bd5a187279c415fb952ac", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "481429af737da0d89595aa7613325a885ca9c417564aab94f5f263e91b90fbaa", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "f92c0324a560c511ea1f5551107f3a04a2d1af9ad9858a7265ac023fb80934e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "9dfe94cea5f3fd10547d105dc8e07b222e108ebfc25419ca03f5b2e3bebbd26d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 384, "7fecfd17989dca983a68eb6e9caf0cd20258f0d0175c424b3298282aea97b077", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1c33181789c6b3bf154b05e055af7eb18d0134ea9471220eba1f21b897de21ec", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "550fb50f86a28413725350ec1418fa82d7d14d5342a8c46a2474b9ae58090824", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d51059ce55c04a87e14f31583234d0e5cac2600d92b163f2398abea36f05bd26", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "cd2abfd79bbcaafe163f454e08a8e7a2a131a728cec9fc30593c850381ac2ade", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "ee6fff8c6d68e42545e74e1c598f3785ef7b4b62475001549da2b216fc74333d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "309ff8d2ae4552e26e8017cbf43ba33218049a3eb6db6af7a0c4aaa3f13b2358", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "35843cbb37402faa18816c655ebbfc83e56032d833456645faf3b8e05df32e68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "26dd7e3104bce5655904fbdfb33568ec512503c8ae5d34b031876dcb83fa2309", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "1d8260c02d48ca71d4d6342498b95533a56bf1508fc13f4e488854085985a9b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "624d975e3baa7bd357171df77f5210efbd61c8df9ce0c2757b0ee604367e6763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "7dcb2febba6e98d99c8802ac116496ff8a0ce373954fdbcf59cf55fde89a0fe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "05f30d72648303522b22400b035f8c62e5e87800f99d4e8db28de0574769c85c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "490804650b9126f9fa82ba198b17f76c6d360f14df3229e56860cfd46088379a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "b64d7dd04b102ce426eda1aae8494fb8eab5b72387321e8fc8b3af67deaf4644", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "c248b361532e8df58b7e5588d0bc8e79db7497873bfd6e7cec2672b1186329f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "e373ceda9556d1ddeef36a87a39e6238ca8a7576285ef8705f4da7cda9089a05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "2382a2bbb5d1a3f01a42133611497b6b85adfd4e396bc9473574fb7f534fa02b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "a62544d58c4f9d437db0962d51c7e530e7903963799c0556b4fc5c7d261df9bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "0228f15254cbf13b5bfb4af5c58a312ec77d8591114ed4c507f2f20aee0797ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "6ba05a210782016b0a19b5ceaf1fee6e22e6106ffdf3fc22688486d953a10abd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "c2b57498885bcbf0ee0bc0c95ef824b56092ccf22e98f09d3dbabd82c10e65a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "b5955686f9a8a6693f3a1b0a82e66f490b7c60664383aaaf8225138d2f682233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "a1e0504524642a773aab1b132b46eddd1bbdf2e4d5f3d8b70fc47c43b1ef5c32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "dd2eb33c07e3f5ec47e8e64fec6ac9467231fc56ed5fb614fa8b0fc8124f0a8e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "56032ad0e4a30c147e18f153d0290ae3fefe8d15b66d7e67f157bedf3f83cf02", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "76d1a472c4242965a691df99b0dd0cd6e75e25ca380d4e142c09e85937b931a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "7f2faf9eb919e90d73a13281ba31eabdfccd1bffed8763eae9a3aff6b42678ee", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "5dd22f6ea255e468e6d84204fead3a5429a645823957082b5508d21a5461f70d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "b62866fdffe99d35ad1db22cbefde5b0b8309d24926f50f4028dd15fc065dbe4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "c2769beb84ea2ba55e7f1cd1c7295e5dedf41f39f667c4675cf5cde4368b1553", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "7b29e7561e4b5d1d5bd439b160a0be0d63a721a22926f2cf293b7a68dbdfb64b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "d4b6a3de746fb3e6020993416c0b399488c3d8514aaa0ced18285ae90da6da6b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "9a6ce5daec4ffe329fc239a3349d9a1a61600d3893e6d07548a4261215f74478", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "13face22b527cf745120bffc9b69067a7aee155c8df64217803dc46fed0aa747", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "6c0b466b6ee6554a2450fff46ffe769fdefd64d97eeacb9cbd9fdb854ee1aae8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "ad716ca64a4b51aba4ee0ba11e0d2c87089deec6842c4359dddf8ebc61db58a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "0734308dba14c60e65f72c49e475eb8e0a1da9441c490961fc2b2780035f6652", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "5eea8a604df9b0f7d8a1ad48ff202f9758ba648215d7a397d2d4a4dfefc2647f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "82e478035c5f4f87966f3d4639a26dd28b5a55746e463f4397b3e7d430c029e6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "07a825a50d397e93139646b94bd43e371763a5a0f6fbe870332191af21546a7e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "28eb92999af544d28bea27ae5b679c8c3ba43153f5ebd5e41206efbe4452781a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "c81919242eb03b52944724d9da0a6f0eccc98aaab4a2d9873e3ceefe231b239c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d59a1c4ab0386ff65b0b08ffb723dd27b07e07282005fd63589842c5bef777f4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "994119ff16048af9938621a6e797000c887d80221fbcddd9d2346ccfc9762712", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "55d04292bf3cc52b68c57a30ba4fc9f99089db1d5b65a77a2cad76dbe16d425e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ee385a713bfde37afb2d813bbb805a678294b5386aebd2f8bbea9e40b39b59ae", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4cc01509589b89e286b1533623fa14c97fb408a35bc9c8fd26b60f5bd51e545d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "f4cdf2801c6cf79c2004fd60e3058821f03695a2e71ad0f9d412b336e2d01a3b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "46fc1bcac964c790efc296de0780687e8e5f5c53393579347d1f57bb9c954c1a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "9a280087e4736bbf6a011a394bf9db3688abbc4cf6c189ad95e0fa8b0df3d50a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "289eda2f78648acc93799d5fe0ff5de277df5b8bdf9a4917c396d70f0400dfe3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "aa4b6dcdc991f8a5d7b389ba1f31a326f13dfef607fa272fae95afc1e49e8903", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d655f137b6a34ef2575a7360afc84196df9c97624841a181e7b052d4aaa53c14", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "205a00e2173453ac770d1bd5bd81f20000bf05a2d95339d692629940bcb121e4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "0c320c6aed97087ac951fb6b1477cadcd44797ba036cc9a8239b31e6b72f5ee8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "547b4247d28e9a76af86592d6277dced87865bba2a62f2c59ebff2ff0d3a4e50", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "bcf29f267892241d4bfed7c8b8f6d538958d93bf41da193078c4d9cda69e9c19", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "05dd16b194f0b034409be3df60231e542c0413b1ef7c2602be35b217455a2df6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "d3ca12a11b50eab1d6087ee98c78cac016a3cdb2fd9a5a8a6fc3671697644c20", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "040ced923e8a8336fc17555b63c0e9dd035afd2e5735f47c94e4a972295159a9", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "8e404fd83fa64570eb08793e412175c466f7f839a58fc5135ee56bdab0641b89", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7033ab1e5a3dc86e453757e2e6347626e0f8e85caababb8d1874c46d63508342", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "95d50d4e8de07974db5aca1c6db7e9544a4c2b77beee0f17f17eeb4698e82a78", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d02542afb60ea7e8535bbd7d0139dd544924de3fe184333c6c93b1f62b576576", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7aca6c519b503d208216b4a258d5bed27b87f2358e04d829d2a9435a3fe4ef9f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "332bcb164521139218143522bb112243852e5d236cc004ef0b85b73d153e0c1f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "673da6fd200672a3b28064b475ad79d28ec122a6c7290174cf36b8820fceecac", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "c049c243b0654178a2a369acb9b9ec48c249c750f72576728264c1aca257e020", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "5a20fbd9dc68ccb789cf0cb59d3b3203842b91aa78ea79aa8d483ee84f11e5fb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d48deb89ba9176573e4fe8dab6d5d201d3bd47ddb818b562f60418ea779d6190", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f4c5ce3d3b4f5eb51a25a7454dc23c800e43ff85f7a62b829b3809e70b718e8d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "72419292ab05cf4d244c5b5c0ae3e136dbf226b0a6483ef218dca3a49b782817", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "aa64e050d286bb301ccdc6231328c6414a3b97ef756d4afceff6769b600ee1c0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "2e90d5e43ca0879a892069bdc2af94f2ea4f5d4d10d3e0590ca56d7117e2c32d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "4e13e6d9e635951ee442d9be9dd7752c2b3c20b8ac236265753c60c788508850", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a7510304576a6804b59fce19aabb2235a315b53c891c86e588c86399fc967099", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a0541cae9bd329194d63e3bde8bb2cd55726259d365823433d75b5d5f5112cba", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "df2740c95229e44bfaab026396c4956e5f56535a6b606d377fa56cb7a73b5979", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "82f4e0bdd5f32fcc54017156419c4020f9a0275d8a2348e96ab9891801e228a7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "bd682319e06e3be668ec60473bfbbf46a2a482e81fe8e6e2c935310bbc1c86c9", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "00fb645c73def4326e39b1879db39b399d21fe57054eb2bac6d9085e56effa11", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d4fc0c3eef6c04d7863b44a4cb475725447d0628758659734dd71204ef628822", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "cc9b01e5d2295bbfb76637153f9421a50b309bef32d38061dc9b4673195e1f50", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7957651d95e91e37935839e322e0ff1bcb469c8fb6a19a90817353d707e4bdba", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ab281f2cd0af8a9dc0d8aadd861535de4608ebc2bc6a6902345284ff3c2190c0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ef55d0dec48c8f1c87d80bdb1376bd8a568b8edf17bc2aa74cb8e5b1ce01a7ae", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "76809871b27eef4603d3845c412d60d7b00eb62023d4c19e5a55594a2ac5ab30", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "5725546f01aedef3278d6d08b1f52f9382ea78f679a07a3f67cf7155253a3d44", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "65e929412434fc6a89a5f4b7bb94e04898bbb611655096e4b63f1a439a3f2be0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "a8e910f07d6c2d217e9840c2cdf5f5b64d2d828fa09f545230c9a7d866b56245", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e784879fb315688eef83798d92ee78d7ec97c0e36b028b252f961c43a2cf49fb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "7257c7f0548142b7a96e1103190621c0939be6ca0d6ac66e9a0c90631ae3be3f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "ce4ad434ede95e8b8b0de3b80711fcbd69c62880db649a70899e2766d3bcd0dd", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e9894541115ef200acd0663750bbc183716422ef15e31454fa96d80f6d232aa6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "02d8157a29ceca5b1ca05903ba2c6b081fe4f0d5c970db65e8fc3b6ce662866c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "8fa41a22e37e979e4296602cf28159b80cd376cd25b14030b9ebc1bcdfc3000a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "45b71c44666768b14c4513caf10799f05d485b5566a49e13693ebda1c4a798ff", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "cdc0de27478128e77db5511b8107bb62e61ea21d59abbfcca6bdb61502daae57", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "4d58ba835db97b51c013e14b0d89e7b929d5d5e8c1521829c5bb3056f32cd6a6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "bfefe86e567501ee9a9f67e5e701df01463b294f3d5768598f01823bd5c108e3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "1f193de351abf614f53577020fa04c9e6376ec007da62264dc3557f23280b678", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "6a845e4de75d53f8f058a23abc3bbfc296ef4a908be10754f4c05cde449636b9", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "646d8b271bbb2330f7d55b0c322b71a12b9db943a8fdbae54bc73788bbc4ecdf", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "86207f022f29a9c4d914582c86f67c38a1ed9b8fc136bdcc5deb935bc115d91d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "b70bc1d9c69b4a7f1a9713351d81f1b4cc61eb055eeb2e9208dca9ffee59d7f6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "36dce94832cc492537865df720cd64a485be427b6364ecba6015b367d0814d83", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "b04b4c88e7f8b3f809d286f1bac1c79809fe2146897edf9906fc7ce969529e9c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ae0f462c644ace475b44d19064ab5eddf83870a238b22102da8c4afc90e3baf7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "4b10629080148d03b6aa6de5dbe2aace7bd929789c180c75984db043820ae78d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "9c25c8364816b2278d5afca83df3d3fd4fa698956bd3108e6da91fbb0504bec0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "cd10b8888a19006ad351e27c97c728d50beac0ae9aca7131864d6502aab2b085", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "b467cfe01a07e751bb8ecf3905d5ba27b20fa13bcb50e2ee4c71bb7b07af0c63", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "62fac92334fcc931a70917a331b8ec63bb98b891d5bf2b8b399a4796363c41c6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a5616d986e366bd2d8ee20f6da388c2d7ca0a9c9ce7416e0acbcdf59ac750c41", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "90c41dd7c53f2fc239ea9e50fb2e398eb864dd1f857bc220a885a05eb0b2f845", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "5724876a195ff4099c892650d96caa195d7b750328703629720c14a2427d27b8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "bab7678dedfc6a5173e0cb80fbb39ad8f44a2bea0e7a1d5e2a837a163d510d63", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "85b5d96ad6f71ed15334c7a8f1c815c23da43711bd6cdee26c5ff827f393819a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "401324d798e54e106a96a506c74514bc51a2fbb0746d8e3261038b2f14956b17", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "e2fdc84d8000cef2db3f172351ed4564675c85b92a1be19e6beab74262a84fa9", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "180e121707df5277d76dd24e0bdefac85b547368d43c77bfa73db092d078e6de", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "32c4177140aa9a36e24215c8378b47988a2f0a8a4b22cc168b6e56c5fb8c7e31", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "6b299a1fdded1ba49a68ab56260b36166a30c5c89c73d8a2b855d291be9ef92a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "e334d43018aa10e686df4b468a711fcedfe06b959ff3a167f49b2abace1d2d65", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "9e02e93d940339eead5d29b4296b84ca9de5c7a0761221d8b1e154a4cb934195", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "cdc0c45fe189f6bcaca72ce4d63a0808f19f00e6aac7b7697207350a9978dc3f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c90916971da59163dd63d7679bbdb53e3a9f8158fa00b57e583187b1a5b1bf89", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "d491c8d6dfb8293df8031ae0423ebf3f20dea04e192db85a82e72380d77bd4f1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "6b0477cac9331b69495d8acfb3aefd18589d54c64e7bb026bd8cd73982695e35", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "1d3e9dc1886c6e850acfd5775fc3123229cf5a552f00cd1964c78d3d968d15f5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "3b59396e5974694c77c1a15ed06f09ff7c0f6f68f81424c55250794bc176c754", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "7f762cb2173b21eacb41d3210495387e74c85fc95a95299e58852306850d8ddb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "2c9b180fa2e56687331fb8be4cedd1447ab6ea4f42905e2b2b869acf13595a18", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "4ce8a6cad6e7781fbe332afe1b66141860ec04587de686df25e774cb701b3a3a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7deef8005709ac01907098e351a59a86981474b86df84d6a223d71a603fff0aa", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "ab7232c1eff57483335ce94be0a8b609e09ce10fa57da1ea207c3c8fb323950b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "82b6f1899e597acb40ac782b952da910062f7f3f5881671ae4ae2dbc8995dc4b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "7f12c4644177c506ad3708658b3ba4bd22d684ea430dc6f8bf101b609e06750e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d75db8dfac7da7b4e78e63c043a48e5bbed5aad60601621acfbd9e947e906f62", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "883929652a073f819f105a2a6383f3ec20060a6169c9019e82ae299c0ff09125", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "b5bca897badbf7f56d9d3126c24f89e64114528cd00d8d212739a4fcb69fd9ac", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "a3f6c7d90862bd7c038e8a50b59824e9ab77b60cca46337da3389f4cb78afbca", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "f1cb620e9eac327f57d6ee4cea10f05e05e8c679e367774cad4fbd3a6a80201a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "74a01c05588b09afe88ccc3a2ae37002fed1bf21a6dc041a09fc03cee7fb306f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "1de10764d2f77a12c03d5cada6d10bb08b21ef0f36243cb50d10329332dfd47c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "da3899c09768f141bb82525e46ffc5d43aa1ce61ed3a59720ef1670f5d467a26", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "dd488b5053c3970912a0dc966ed2536f0fbb1f6702291d62be49ed32a697bb1f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "aa94c31be056850d2ac6688b6626034f08cf89fb52e34b14154a6e91acfb48db", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "da1b9ee4aac7fb2e1e349ace13d830f54cdd61374a202e784e91325b6da0e801", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "41c8c4fc1c1198803e75f65ba1797676637185ce7ec63c3b59d67cd5f06e5e2e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "9932873ab42427058acdaaa2841bc915686a6e4c8098631bf088e5e736f7b666", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "eea16b9c005509471f769ce1fc614d76e8e6042549f2950a8ab59178cdbaa44f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "bd4bb30e58bc2302fabaea8edb69f0e5130b7548829511b7e666a68837d4aa9e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b6ed648638c289f1541c0cfd8a9eadbb5fe81e655784b292498c489dbfa7aabf", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "456439fff7d3d1a475eb874682f8a2b957b2200a69a35f1f6a2a77eea8a819d2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "efd7298a961eaa8239e09e569a6d8be859d9c066a0d462b29a3bb9813a06ddf2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "91540c4022446e233208a4e2c99a5218fad4c14104d243b7a6a7958f3d0f39d2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "966d733c5c8842be5927203bf6eadb6b55cd13091d56ddf7478e7b2d1c80ebe3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "33d889736c2dcfb26e592e4661818654a9d9dc68ec45a6b1656c538c372360d5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "d3eb9ff2a0757f7b11e89c42573a42e5da7267bd77fa89a6806db5100f717502", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "44d4e826390127f72aef4ff121cc4bb6395d5530480be7902056d151cf3fcce0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ee28ad5953c95f5bd77991121e44520c574768caf51cf452f570ee73a5dbee04", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "8e158f61b37659935e2fa119f289c94a694b665de2e631203e3a3573f065bd5d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "2fc783368b0e954ef9debc9b1045f7205ec2480e63b9ad6d11793b6ddc161680", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "2e2ecf3f1acde97fefbee3c4864f69beee6c6c1157d0053fcc17d66a244ac0e0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "1fb468da7179a53978d2755a97b51eef16a5074a51136751d3c63683ae49829b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "60a0e6627ed5040c8d25f6cd22714c19f85a274de6dcf4c55a356b0082b53982", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "15ddde3d398dc838759849f4950ea77375f6d4aa0999b343cab83575ab7961cc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "4e9fe2ad3616e6fa55b85ed069e93272ba1ac81db3ccb52aff5ee5249791994b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "d87d8f0af0861774fcc6df6c7f0181a40ae2fcb61bbe494120dc6b3af6249076", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "e607db6e779702532a097336adbec929a94b9689f8866c0dd4fd1001fd259b67", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "5ae37746b50a42b9071087f564778e4835bd9aff2b90b41bc533827d98da8401", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "f45e41f48d8da2313b8e487d2deeb9fe9325377fbecb9f07b4ce0585a52d4791", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "b077446bfbbe55632d7f25bb3d2d8353abc0b6146c14ef32fb17e1d169fe8a4f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "a732109095dd05605d980121a78df7dbe0a7011409b9ab369d172c225f5e65e7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "35d462eb60ea4becba625bab468e6f12854d3d3fed4805ae571776974aca60f0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "ba2d2ea06c6461769874f571380afbdb767cf4de35fcbec7a6424f4ed109a476", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "84252e6bbddb4c9fcc520b877b19299d56ecfe63413a574783c5bb7116e31c78", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9afcace2f5c1b2245d5d0570c4c2c7f120832d7ede7eb44dcffc2d0205b64855", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c6f82eec0bb4d099fc6291082f7df4b39676456627a1866a4715d2f53167a0bb", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0283b2f3b1f0234639b70577bd0e4ebe26ce17c46301e254d6857057e78c115a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d8a091d3c50b2eeba90cf42f8c973b7b1d2c9590f46c236abff439f3755002ba", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "80ebd9e645a9e485d30f31fedd8e342656e59787e2a2557a4c338722a889e842", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fb48bd280e4c7e954960c7841be8e6888974c34279c00ec0e14bafe4f76df15f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9ed766d1113b1ef7ed12c59d4345c37bee6a9cebb5dc96de065cc63ae7a829f4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "29185f730cc5e302d0edacc5ed821c12b5cc7ba2bc8febef0f885ad0f553e869", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "967d3084e0fe15f92e06d0b1309514c9d7c114af9764afdbe7a45f50242a318d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "12d4340145805c05d63e308c8547ff450496c51664f3dd9062c6e7f511bef239", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9cd0e97b0ec2c8f7fcf6106078aee7900b07b90fb499c9fbea48589060b5ceda", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "949457dd31541a164aecef314bcbe1e18d0cf98808e1cc01fb0656e05a353457", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "15cb41a3948c12b2140bfeea26b5e491d5980f0913b4902ebdf737f095e2382e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "4aceb76fa455ff9f74b0ef4fa0fb0d4031e94820e74c07cb17c95721788a976a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f6673db01d2ded4aacb6ae75e6d439faa9707d9cf398a26506905cc6772d2f52", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c20a0b7f189f63a744dea4932289d81cafaba80a4335a42338305a8a7dc18e1f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1736f44aaf0eee2b5e61a8955677e02419e20ee17faf1e4be180a30f5db53795", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ca26f4bd629410476d95304b4288e24969b0938968a37ae504e24c508fb1fd6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aaeeacd313389d6f217bd829e73c14198883be420bd85ffdea2204dba8066cb4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "731072195eaccb27f16976b55faf192569ca9374b229d088bedb3c66112afdbc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27a96bc45bab2c083c489140b5155e45a7047202b007a220ca51d9923ac54a0f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f9540efe1efc323f46998ae502db922b02c1c95a6077fa2f6413600687b73c39", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a6ce1726127f1acfb13598099b86d3ba9f3b7831fc344ab5022c64fd4530938b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b7de2b6abf1014e736694a7731267822f44ff52ebf44d323873adf3112613a42", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "30d1b4f5e539e93a3ec5b5d38dd39efab8d75b612e63ea3418063b86cfc72758", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "02116cebb86be804a8f45969c3eb0a24efc0a2e2b0a7337792d7bb99df7674b3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fbd399a136154dd0e7850bed4c5895f5914c87ddbdf04dcc355844f8574ede50", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "ccc4faf4893237c3aced7b8886a3d509e02847211664db9c47c9fe00415ef04a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c445fe246f8fdcd851c441b98f188796c5929d52721e9e1016da873b6528ef00", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c0d56d15166d92a5d9bc3f782fa0270d3920ecc4fa4beb539060ba651454b419", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "283a1a91335f9d56da0d2618f274ca45cbd583fd19edfb4b6d72e3c6c603c18c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bf4765d2d05c80b44e29e966c84d66a8cd92db370f4954a757938360302ed86c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a6853c0d2bd6819b1027ad30586688ec14096def014e679e48f9e17b0d0f0db", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "98c3dea165b803cc5dcbd007a791b9511a5b1a9fc12831c090eead6c3e616159", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a687dd10c4aadbb1c5d8384f84f1a21bf2902854f986c9b19a47d9ca48180cbd", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "85c09f253b6bba56edf746b9467e602ba50deb63b4aec00f2e8b8993a3717e1b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2297ac36888353ecb48eebc5c53919ca2dc7d47bc33772fe137f394eaa2df51", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "d758d1ff2ecdc149a1d6c01b349619e5963a3b3fd3e3f269176fea66af3a76a4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6da9008e57853c0359f650abc8705547f3c2acf65eab997bf6bb4a0b69d27d81", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "513ce7f61ee29ee81fb566da09f920b3ceaa4f8c25a637b0a21c2fa091b37ea2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5d5ed6dfb62cf68813b88fa925cb9e96cfa4395ebc3f6ad6aea3b9da4e9a51e3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "5bfbb160eb061b53f7a253424ad5f439cafdd1a2d66646fd69d364f714889827", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "737c22d3f7d2a7df1f780197552ecafa355f3ab257192eeace355f089f6e36b5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "46e4cff50b37d09bcef9e8786244aed2ef1a7d11798245934074a198507db5c0", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ff4094890167632b361d0b0e13cbf20366048f0c549d8c33d7138a30dbe8e74f", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8acda73509af9875b5712d59ea91f287a6af2eef80215c4afc666e62e1f18478", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "2a1cf6cfef36beeb6775d5f3c1799420d1dbba8d9caa14ccb3f64e1090e11f25", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9bd1b72359b973a127ec5b9949348c89940edac95ba27885c9ca5b4514b6b297", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9159853f8827041b6363a7756d2511241c764335bff0103905b5a72b84fd138c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c406931a205cd1a232f0c690beee37f3b8436265689ea5191e4f19b908d3ef9e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "24a7ecf88f285d545180c8e6940cf35e9e58bcccf225cae9cbcdc5a3da13a65e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "a35568d6fefc4b8fdc4b280e5298a3f1a057a197def9d4016f42575636464339", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "008cccfbb923d53c2029a33aec651b574140ac940c95c5049323bca6fd71048b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "eb4d3a4d8279d3d61c6b3cb5f9b6eb57282cc50a09668cf2b22a4934fe438235", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "dcfd169a19a7a56688848cc3e530cb50a633f9bec0dd1f088a3affbaffc0aa32", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "90ba448b426836c4d497722567182b5e56f836e865001b1703ce3c33d72c5f89", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "44b7894259c7c6e2ecca1188e75b8a6d28fe15c1d6a7de5320cf030b2347f1a5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "0704c95ab7bfa673ff914040a2bd86b91d0144942b4e421549314f50a4d6311a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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, "947c988410b4d03580f7d6b2a4cd3a9f253d4798585b82dccebaa7eb669f82dc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "324a79e9bd34da899319693eb4542e9743eed79ae755eaa912b6759df7d78899", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0588fc2c6ee4d97da9b3c0769689b88b30c957995fe277d610916e464a703e6e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e4668c5466e5b67407adffe15026008136d6f21b60eb6caacd4c63967de2c515", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e2756987b7436c7d26180815d6e90a4f4e52d55a44652596fd77353f8ff5081b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "537b0a389c0560268bbbca5318cdcc9236c8fcb66951b9ceb1e272e39fd0fd07", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "e9ca36d721153f485de1564f4f1cb48e920c823234d6f532382b0582249fdcbe", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8818567287ba55325675cf80e6272d75b0658dffbbbb73d1d3383d5c9d06b374", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "540a545f32a755693a5e9f6fcc389bfb85c935b3a2259b0b1741773e77d19951", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c76c70ac24e822564af646b857a320aaa8766648b3953c843c680e23f2fa1e31", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "13854c2b2f56f1fc2a51a3bcdfc1b20c460c07fb41cb6f985c496ce74521db0a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "017624a1b3d5863292718a6f024699921d5c37d28b5c2d63101ba5e785fac085", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b5b88d96bcd942c92d90309fe56efa4d57893f7415301ab9a51985b0b979fc64", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "f02481ebba149c1d82ebd193f1e08b66695a0a7745cfe85756e145a56a7672e1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "be7445a4b7c37153af7769d748f7b1e2c22a912692393efe7dee65ba79c296e5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8885c7af1607d290c6db961b44ba2fa474b71393b31cc5ec84c65785a5f94fe4", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "68c78c257fcfcc901f83b7d6e3a601f32dec86ca82d6508fd649ceb8cabb1a88", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "667b4f668e86247a8565107f9e5c891732a804240a8e3784f50c589a390a6dda", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ba0dda2263342efbccb65780ee9e153360725c9c60f0e7be6e23c3f16886aa0d", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c73b5878c682decfd4c4be3d85134f8807d492eb7524145db7352115ed13b2ce", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "235e6e6fa4560c8b37c0fd7fde6a00efc2f23a9ecbefe4de5b0c97b1ba301495", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "50bcc8a3b1a5899ab68c92164420bfb7cb219756ca4507d1e5faf08e17a9b598", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c60ad685f588fe9475d1cf2a279426174a868de370fe998f1da5f90330a18e75", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d341e88eaf408c66d7d7793cef8f6f6796969eae744b811cc70c32df12fb7c1b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "68880b3686f8f98419f4c7ea1685eb3a903272ff2297439349d4cc7e55525da7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "221b8ac113c0b7c940fd730105bfa132794c791057756fe3011b2e25a0f81f08", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3121cf04def03564a5d5ab78bc702e9f6150fef9e18984b8ce50fd57f493ab0b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "becd5410f2c44313835aff71e7cc410e8d82884471cd735713a5b5185faebaf3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "052d353c2f1f270df79a89808eea0b247861c4fe7ffc1f609b1fa591c8225705", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "b07636f83cc76c97c8d7c9d0776b6db4267fe83dac46edf676ff78473ec3aa58", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a12068c6f556b374207854566fa65909eb06f131daa7e38bcef4401219c82978", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "a53322e60a2a88aa92a005aeaa379ab9019323092e2b0bffeff3158286d75ecc", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "230d1a4c6596fb6c8f117d5aa09a9f967f8bb55cc3fbc152722a377751e87c5a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "7aab89c1173f671e85fdb9bf87977fd9f8d85867a6f49529d68f0ed42e51eaa6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "7f123072d0b14b572efb321b25c4c87e3a1d79f0dceee2e27ce073703fc15a71", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "011ccef232ed3866736f7de408c543a5166b8aa0fb7bb7eebc7a927042154a03", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "50a16b2f1f06fb7f7af9ada651791e6c13134bcd8dcd8db972d534bdf6407bef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a48e5c04fad94fc68977db6585b1fde90268c40672ce5a025d335f60961b284", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ab8eb3616cd572677ef624c7254f54e9a46fe586cbcbf5a43bfb5ec2ed4f2f3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d2ebb05f383eff20f036de130b18a0c0061f07ddc3ce5f2c646e8d074be6300e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6dc659480c8ddf4105d2474f4cb45742f1c5ba99e86402b36b330478cf0fbeec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, "9952c871412de20a6c8ee760f5df798ee171ed7f839c1ba57c7336df6aa96610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "462e39655da2943ec9381f6780120008ecbdc4ddc80baecf769f26734be72ba0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d3657df8ad4aee0709a705a24950effaf313acd94b30839b216daceea8ae934a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b193a843e4fdfefeb9ef8a8c47f2b770204bea68a6652c75055d7d7dce71f01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "842d07154a2ebcc7dc8a4bd5f7badf1777bd6b9c8e64f19028066a9a053dfebf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3f2812c8477ce4746d46a67440d4810c8356136c292cf8a8d05aa45161fa1948", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f871afb22c34eed24c7b8af8a2cc6225f579bc2be21505ba672dd925ab5f4801", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7f8cd267eb7d0418bf79a6c3ddaf43a387ec96077f12a4ef1dc1527079f145f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "b1268a5df8257db5be80412ce3d161c6262b236fcadfde5d0a4eeb460510c534", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "461eb661c76850604ab06696eb6a014046da8c642690f7b7f13be03e9cc6d6b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "def3b757303e296de6138842b0933611aa988a486864711cddd6818405bd5da9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a154a597ea43dab35726a7c94f0b8286cb69393ab2c88e10acd73b0aa84df7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "78205375d1db2cbfacab0f7ed50b9384019f2bf84bf75f29ae0331f3c098eb59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "2ea431f5231c7e948e896956341173e7e9264a6128bb73e5479e8de2c0522920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "a9291357eca60b4a0a0c0be4f2cc79fe994c90a10ce3fa0653a555402653a60b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eb9387fdeeb9ed0bf51e1d63fdd43f20aa26913b1a3605ba1f34b9059ae7d94b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_rgTma_clmp_dynB_sm100f", 512, "8010844f7e60e625f755066f59510cc973561da366be25d5f53f89facd0e5a29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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, "cada83554da37a7b5dbc1c10738a592fafb2a3de384e646f72d9e536c4823a9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2cec70fb85bbecca3c49ee4afe99f3340eacfdc3a2c2bf2d45174386b46c4336", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "33ae0535249ed98cc1ae3d840b090754a159368df7803435532f261442cca69c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "dd1ba17da316b86d051dd0eb0e4eaa92c02e3213d004c572304dc374a4c6ad6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "d4568bfbf6df961675115a16692f7fe2ce6bcdc1f8d7ef02d17915261224367c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "92ef64c9dd9ca4fdb9b5d325dd4730dce7c6644b8c74fc0680f251e43eb11b23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2eb73711121fbe25f78f182bd3cbe344ee9bf2a666318c7ae92e816480a9919b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "d188e84d76b649d7e6455a89c8a6073912cb133258456105f8e69ca3e277807e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "2868553f9915d026570598743b7a0f0d6c4184c2865f80555fdd2bedbd7562b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "f967456bb7da362012d47745e0a2d51ada01102b2923d64bef28e5d1daebbdf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "8df0f24a72c7205003eea261f4dd4466df7d2db87405881e554533472e734d97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "96258ca54194bbbd19b649ab90de66495a6bacf383951abcb9ad53d045611c05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "96ca02fe410b780ee54760b1d0e7950657cc911a13873ffb47c7ee33a3cc0aba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "94bb81f411f7fb0c343a7449bc8b78f5d89b540bcc825b5f86d8a084900624a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 512, "318a51217ef09e992fe06e078ffcb63558c361c984a06ec75fd254a0e15f5052", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "49280628610fb9715af7b786ed76e4ae6b0ea4dee7b7b143908667de23251dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "48627e251682f2d54807e5fe3213a48c30f5e94bc69a3cefcca97d51e412d5de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "a4b1671ea6537649591fb942282ba4a1f5a4579584455a50fe8b4213e2ddfc6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "e411829011244ce821aaaf799bcd60a1be672d97d990ae51a76d0fff7ee95f8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "07cdbdf914f43a9323176c8f9410927a03cfab8c86708d5e5e0e81783b15f02e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_sm100f", 512, "d02a8ae49aeb830eca148579bc70e34ce85df0779a64d1c2c49d5564b9829067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "085da143a8f64d422691b0135d4419a9ea4253a32254e0fe353039fd6fa7f37c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "1cf29f498ce4ab7f889c506435e6dad302a1be71655caf05b8bd34129c176cb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "f19ea6393d81f65b396d45b3e55b285b0342f8276bd3a5f85eaa7affb4e18da6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "934025364b736e736f1063785e7737abbb2f8c25073daf9249561d1ed8f46382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "351eb331a8608f5f7913f1f9549ce9567ee889d480ba3ecac6e66d6c105b46bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "817d15d20ed1f70c15ac0b762580e196368f1319e336b360480d8817127f993a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "944cdb514b2ffec20aa7037b6065b58d65906fbe815db22ed0e1daf034087a0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "5c91e6ff23fd19cafa9824fce20592345b8553f2dab5f844eaad3c571ecfe6db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "aa508137d86c446061a6767406a3221328cda3159e41fd843031aafabf10dcdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "c183155a8343679c613b3a08f02bd2263e58c69ff6dcb385710071e180b745f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "1a93d4b6734eb0048f6a3c26af9bc2b92fe9a7d02f700600c61257307c234b5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "ea09f17083c9c3fddd1b23c241419a396a70b32373dd7cfd12ad0fb6b6086e9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "771d6a00a02a29985e9cb8e97985dec5a577a32e57a637210da8b1b60c8a3981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "20258e819937ab8cd6dd3d98174a1cce36fa9550a2142b6f2fa2b9238e3586c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "393d64977d41aff2a6672a7f247ef64483ba325ef9544423c6484a46785ace80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "92b13e9e4267166776cc5b4420f29a3b8d67cb1024812ec67894f308c9aa1ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "2f0a4dcf13b5ce856c3839bcf4115a9617a4692505abc641a5b29ae4af82d3bf", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "23a1353a5c65f39eaf9a0aab34f1528effc969839293388aea6c8700f2c55a5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, "1a1f6bfff069a778f0bdcba80e1d77ccd09d8f5675a17ed0d6fe580c7834b2a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6cbdf0a59344f5e24b0a4804cdfd1e389f8ed7d6c1dc10711d575006662c5ea1", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "195ca20635308b8572de5cba7a8a1d2cb570af1d7a78a1d88f2499137eb7aaa6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "c1eaf3de6795562e6a14f3962d27efb80e679399688ebf59c4800e0acbe6cda6", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "d787b182fa3eadd71ee426d0530271c1b168e689d22be48b0cec937eebe6f359", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 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, "7bf6c6770afdd10a7410acb19e60bf74286b6d36d198ec7a55ebee26dbd8c5e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mSliceK */ 0
, /* 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, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "cbb3246c89a6ea74ae2034d926811c911df099c99db931afecc2df1554872111", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "bd5e689844764bfa324f40064823574194f31131108a3134d14aa975ba7cbb6b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "606c2756a42be8c47a9a0bc2333c28aa7677af0b66cc229b0d20940b900f6725", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "063712a05c9cf782c11a5e49a8c5d3ba63754f5f2f7d6e44b0de6cf8b50117b8", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "81c050b4b8a38a56f7fcfa5fd52ec7564254a4d4ac84f9faa3ab970930111233", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "852d72f2dcc8d2f8f7c77cf60168e334601290cd007e54cfda430518d4cfddf3", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "7692cdcf0d0133fd44130115475c91ba2bce254b455950bd009b615ea6826544", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c1237d2b3ef675adc7742e00f42c5d3d6ebd4fd78f193323272d26167ca8b497", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "67c1c0554cbbe95fba393bfb92fe29da3a802f145e4cf4c5639202ddef34c910", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "4437d961901f5e56a53825e3b6ea611902d3f6068626da170b632b3c78ec0f1e", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "7bbba6840ae66ce22648446b51e123dc1f51097da0a4a3477e9071aa42bedb34", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "77c06b38878cace23c4cff787b189bb8134dd94f2e3f46716bbed7a64198d68b", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "2f9af32e8f855e4a57abf9e2f06675f110874fa204b7a47a66439e0976a43538", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "5274afc80f5d9276486d2cdcb08b75316a9779a81a73b3e2c0675cb431f3fdb7", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "b25957de82657d0e32dc1411e94597a1dc5d14538fd3af0a568671aa26c7c438", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "7d08b27b4e5ab489b91fa41e8e469b6b538b2664bf78ffc8e829780b0c0dd337", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "95b479aa0c86e505fea41e3cae06bed8700506f63f917781ebd110166c7ff8c2", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "0a2f79a8f60007abe0943e4e58496bbf284823c8a9bbe1933ab3f4a061a7d15a", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, "32d2e49b7da263b1a4a9847be84ae100dad3234b21ff3d40f49eb5601db2127c", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "ed9b6be015d6ee0ea7abeeb4db0874ff284556d3097e12977da09fc83e34c7a5", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "aad8890215f2502a6e0d4606e55f4e73e2f9bd4f85eea1c92c51c2704fc9a135", "", 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemA */ 0
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSliceK */ 0
, /* 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, 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, "6e6b809c4fa6d62abf34f5bc12f490a957bbbd0359fab4f1ecafa61b7e467d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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, "3df451fe09c811d457c64c5ecf26c19f386415b4f564d5595e318a7280f51c0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* 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)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* 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
, /* mNumSlicesForSliceK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemA */ 0
, /* 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
, /* mSliceK */ 0
, /* 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},
#endif // EXCLUDE_SM_100
};
// clang-format on
} // namespace kernels
} // namespace tensorrt_llm

}

